another fix

databricks · Dec 19, 2024 · 6290691 · 6290691
1 parent c48fe48
commit 6290691
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/providers/pluginfw/products/serving/data_serving_endpoints.go b/internal/providers/pluginfw/products/serving/data_serving_endpoints.go
@@ -77,13 +77,13 @@ func (d *ServingEndpointsDataSource) Read(ctx context.Context, req datasource.Re
 	}
 	tfEndpoints := []attr.Value{}
 	for _, endpoint := range endpointsInfoSdk {
-		var endpointsInfo serving_tf.ServingEndpoint
+		var endpointsInfo serving_tf.ServingEndpoint_SdkV2
 		resp.Diagnostics.Append(converters.GoSdkToTfSdkStruct(ctx, endpoint, &endpointsInfo)...)
 		if resp.Diagnostics.HasError() {
 			return
 		}
 		tfEndpoints = append(tfEndpoints, endpointsInfo.ToObjectValue(ctx))
 	}
-	endpoints.Endpoints = types.ListValueMust(serving_tf.ServingEndpoint{}.Type(ctx), tfEndpoints)
+	endpoints.Endpoints = types.ListValueMust(serving_tf.ServingEndpoint_SdkV2{}.Type(ctx), tfEndpoints)
 	resp.Diagnostics.Append(resp.State.Set(ctx, endpoints)...)
 }