Skip to content

Commit 66da73d

Browse files
committed
code changes for the transformations as well
1 parent d8e766e commit 66da73d

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

ads/feature_store/execution_strategy/engine/spark_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def get_time_version_data(
4747
version_number: int = None,
4848
timestamp: datetime = None,
4949
):
50-
split_db_name = delta_table_name.split('.')
50+
split_db_name = delta_table_name.split(".")
5151

5252
# Get the Delta table path
5353
delta_table_path = (

ads/feature_store/feature_group.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -432,6 +432,11 @@ def with_transformation_id(self, transformation_id: str) -> "FeatureGroup":
432432
FeatureGroup
433433
The FeatureGroup instance (self)
434434
"""
435+
436+
# Initialize the empty dictionary as transformation arguemnts if not specified
437+
if not self.transformation_kwargs:
438+
self.with_transformation_kwargs()
439+
435440
return self.set_spec(self.CONST_TRANSFORMATION_ID, transformation_id)
436441

437442
def _with_lifecycle_state(self, lifecycle_state: str) -> "FeatureGroup":
@@ -1050,7 +1055,7 @@ def as_of(
10501055
"""
10511056
self.check_resource_materialization()
10521057

1053-
#validate_delta_format_parameters(timestamp, version_number)
1058+
# validate_delta_format_parameters(timestamp, version_number)
10541059
target_table = self.target_delta_table()
10551060

10561061
return self.spark_engine.get_time_version_data(

0 commit comments

Comments
 (0)