Skip to content

Commit

Permalink
[FSTORE-1411-APPEND] On-Demand Transformations
Browse files Browse the repository at this point in the history
* setting transformaiton function argument names from json response
  • Loading branch information
manu-sj authored Jul 25, 2024
1 parent a607317 commit 85105f5
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions python/hsfs/hopsworks_udf.py
Original file line number Diff line number Diff line change
Expand Up @@ -713,6 +713,14 @@ def from_response_json(
if "dropped_argument_names" in json_decamelized
else None
)
transformation_function_argument_names = (
[
arg_name.strip()
for arg_name in json_decamelized["transformation_function_argument_names"]
]
if "transformation_function_argument_names" in json_decamelized
else None
)
statistics_features = (
[
feature.strip()
Expand Down Expand Up @@ -762,6 +770,7 @@ def from_response_json(
dropped_argument_names=dropped_argument_names,
dropped_feature_names=dropped_feature_names,
feature_name_prefix=feature_name_prefix,
transformation_function_argument_names=transformation_function_argument_names
)

# Set transformation features if already set.
Expand Down

0 comments on commit 85105f5

Please sign in to comment.