fix

databricks · Dec 19, 2024 · a1f3439 · a1f3439
1 parent 6290691
commit a1f3439
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/internal/providers/pluginfw/products/registered_model/data_registered_model.go b/internal/providers/pluginfw/products/registered_model/data_registered_model.go
@@ -42,7 +42,7 @@ type RegisteredModelData struct {
 
 func (RegisteredModelData) GetComplexFieldTypes(context.Context) map[string]reflect.Type {
 	return map[string]reflect.Type{
-		"model_info": reflect.TypeOf(catalog_tf.RegisteredModelInfo{}),
+		"model_info": reflect.TypeOf(catalog_tf.RegisteredModelInfo_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
@@ -76,13 +76,13 @@ func (d *RegisteredModelVersionsDataSource) Read(ctx context.Context, req dataso
 	}
 	var tfModelVersions []attr.Value
 	for _, modelVersionSdk := range modelVersions.ModelVersions {
-		var modelVersion catalog_tf.ModelVersionInfo
+		var modelVersion catalog_tf.ModelVersionInfo_SdkV2
 		resp.Diagnostics.Append(converters.GoSdkToTfSdkStruct(ctx, modelVersionSdk, &modelVersion)...)
 		if resp.Diagnostics.HasError() {
 			return
 		}
 		tfModelVersions = append(tfModelVersions, modelVersion.ToObjectValue(ctx))
 	}
-	registeredModelVersions.ModelVersions = types.ListValueMust(catalog_tf.ModelVersionInfo{}.Type(ctx), tfModelVersions)
+	registeredModelVersions.ModelVersions = types.ListValueMust(catalog_tf.ModelVersionInfo_SdkV2{}.Type(ctx), tfModelVersions)
 	resp.Diagnostics.Append(resp.State.Set(ctx, registeredModelVersions)...)
 }