code changes for the transformations as well

oracle · Aug 9, 2023 · 66da73d · 66da73d
1 parent d8e766e
commit 66da73d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/ads/feature_store/execution_strategy/engine/spark_engine.py b/ads/feature_store/execution_strategy/engine/spark_engine.py
@@ -47,7 +47,7 @@ def get_time_version_data(
         version_number: int = None,
         timestamp: datetime = None,
     ):
-        split_db_name = delta_table_name.split('.')
+        split_db_name = delta_table_name.split(".")
 
         # Get the Delta table path
         delta_table_path = (

diff --git a/ads/feature_store/feature_group.py b/ads/feature_store/feature_group.py
@@ -432,6 +432,11 @@ def with_transformation_id(self, transformation_id: str) -> "FeatureGroup":
         FeatureGroup
             The FeatureGroup instance (self)
         """
+
+        # Initialize the empty dictionary as transformation arguemnts if not specified
+        if not self.transformation_kwargs:
+            self.with_transformation_kwargs()
+
         return self.set_spec(self.CONST_TRANSFORMATION_ID, transformation_id)
 
     def _with_lifecycle_state(self, lifecycle_state: str) -> "FeatureGroup":
@@ -1050,7 +1055,7 @@ def as_of(
         """
         self.check_resource_materialization()
 
-        #validate_delta_format_parameters(timestamp, version_number)
+        # validate_delta_format_parameters(timestamp, version_number)
         target_table = self.target_delta_table()
 
         return self.spark_engine.get_time_version_data(