Skip to content

Commit 955859d

Browse files
author
GnanasundaramSampath
committed
[azure-ai-ml] Fix hdfs mode handling: revert dict binding str() conversion, fix mixin .value call
1 parent 756f075 commit 955859d

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/_input_output_helpers.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,8 +282,7 @@ def to_rest_dataset_literal_inputs(
282282
input_data = LiteralJobInput(value=str(input_value["value"]))
283283
# set mode attribute manually for binding job input
284284
if "mode" in input_value:
285-
input_mode = str(input_value["mode"])
286-
input_data.mode = INPUT_MOUNT_MAPPING_TO_REST.get(input_mode.lower(), input_mode)
285+
input_data.mode = input_value["mode"]
287286
else:
288287
input_data = LiteralJobInput(value=str(input_value))
289288
input_data.job_input_type = JobInputType.LITERAL

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_job/pipeline/_io/mixin.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,8 @@ def _input_entity_to_rest_inputs(cls, input_entity: Dict[str, Input]) -> Dict[st
233233
for name, val in rest_inputs.items():
234234
rest_dataset_literal_inputs[name] = val.as_dict()
235235
if hasattr(val, "mode") and val.mode:
236-
rest_dataset_literal_inputs[name].update({"mode": val.mode.value})
236+
mode_val = val.mode.value if hasattr(val.mode, "value") else val.mode
237+
rest_dataset_literal_inputs[name].update({"mode": mode_val})
237238
return rest_dataset_literal_inputs
238239

239240
def _to_rest_outputs(self) -> Dict[str, Dict]:

0 commit comments

Comments
 (0)