Merge branch 'feature/featurestore/1.0.3' into featurestore/stats

oracle · Sep 21, 2023 · 4f1606b · 4f1606b
2 parents 96e8ed4 + 05d5032
commit 4f1606b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ads/feature_store/dataset.py b/ads/feature_store/dataset.py
@@ -865,6 +865,8 @@ def _update_from_oci_dataset_model(self, oci_dataset: OCIDataset) -> "Dataset":
                         features_list.append(output_feature)
 
                     value = {self.CONST_ITEMS: features_list}
+                elif infra_attr == self.CONST_FEATURE_GROUP:
+                    value = getattr(self.oci_dataset, dsc_attr)
                 else:
                     value = dataset_details[infra_attr]
                 self.set_spec(infra_attr, value)