use sdkv2 for existing resources

databricks · Dec 19, 2024 · 01231d8 · 01231d8
1 parent 2c088af
commit 01231d8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/internal/providers/pluginfw/products/catalog/data_functions.go b/internal/providers/pluginfw/products/catalog/data_functions.go
@@ -40,7 +40,7 @@ type FunctionsData struct {
 
 func (FunctionsData) GetComplexFieldTypes(context.Context) map[string]reflect.Type {
 	return map[string]reflect.Type{
-		"functions": reflect.TypeOf(catalog_tf.FunctionInfo{}),
+		"functions": reflect.TypeOf(catalog_tf.FunctionInfo_SdkV2{}),
 	}
 }
 

diff --git a/internal/providers/pluginfw/products/cluster/data_cluster.go b/internal/providers/pluginfw/products/cluster/data_cluster.go
@@ -41,7 +41,7 @@ type ClusterInfo struct {
 
 func (ClusterInfo) GetComplexFieldTypes(context.Context) map[string]reflect.Type {
 	return map[string]reflect.Type{
-		"cluster_info": reflect.TypeOf(compute_tf.ClusterDetails{}),
+		"cluster_info": reflect.TypeOf(compute_tf.ClusterDetails_SdkV2{}),
 	}
 }
 

diff --git a/internal/providers/pluginfw/products/registered_model/data_registered_model_versions.go b/internal/providers/pluginfw/products/registered_model/data_registered_model_versions.go
@@ -33,7 +33,7 @@ type RegisteredModelVersionsData struct {
 
 func (RegisteredModelVersionsData) GetComplexFieldTypes(context.Context) map[string]reflect.Type {
 	return map[string]reflect.Type{
-		"model_versions": reflect.TypeOf(catalog_tf.ModelVersionInfo{}),
+		"model_versions": reflect.TypeOf(catalog_tf.ModelVersionInfo_SdkV2{}),
 	}
 }
 

diff --git a/internal/providers/pluginfw/products/serving/data_serving_endpoints.go b/internal/providers/pluginfw/products/serving/data_serving_endpoints.go
@@ -32,7 +32,7 @@ type ServingEndpointsData struct {
 
 func (ServingEndpointsData) GetComplexFieldTypes(context.Context) map[string]reflect.Type {
 	return map[string]reflect.Type{
-		"endpoints": reflect.TypeOf(serving_tf.ServingEndpoint{}),
+		"endpoints": reflect.TypeOf(serving_tf.ServingEndpoint_SdkV2{}),
 	}
 }