initial commit

Author: NitinMore7

internal/integrationtest/datascience_model_test.go

@@ -165,6 +165,7 @@ func TestDatascienceModelResource_basic(t *testing.T) {
 				resource.TestCheckResourceAttrSet(resourceName, "created_by"),
 				resource.TestCheckResourceAttr(resourceName, "description", "description"),
 				resource.TestCheckResourceAttr(resourceName, "display_name", "displayName"),
+				resource.TestCheckResourceAttr(resourceName, "model_version_set_name", "version_set_name"),
 				resource.TestCheckResourceAttr(resourceName, "freeform_tags.%", "1"),
 				resource.TestCheckResourceAttrSet(resourceName, "id"),
 				resource.TestCheckResourceAttr(resourceName, "input_schema", "{}"),

internal/service/datascience/datascience_models_data_source.go

@@ -265,6 +265,16 @@ func (s *DatascienceModelsDataSourceCrud) Get() error {
 		request.DisplayName = &tmp
 	}
 
+	if model_version_set_name, ok := s.D.GetOkExists("model_version_set_name"); ok {
+		tmp := model_version_set_name.(string)
+		request.ModelVersionSetName = &tmp
+	}
+
+	if version_label, ok := s.D.GetOkExists("version_label"); ok {
+		tmp := version_label.(string)
+		request.VersionLabel = &tmp
+	}
+
 	if id, ok := s.D.GetOkExists("id"); ok {
 		tmp := id.(string)
 		request.Id = &tmp
@@ -345,6 +355,14 @@ func (s *DatascienceModelsDataSourceCrud) SetData() error {
 			model["project_id"] = *r.ProjectId
 		}
 
+		if r.ModelVersionSetName != nil {
+			model["model_version_set_name"] = *r.ModelVersionSetName
+		}
+
+		if r.VersionLabel != nil {
+			model["version_label"] = *r.VersionLabel
+		}
+
 		model["state"] = r.LifecycleState
 
 		if r.TimeCreated != nil {