Skip to content

Commit 135657a

Browse files
Ruilin Chenfacebook-github-bot
authored andcommitted
Fix ph_key issue with trec pipeline
Summary: Fix ph_key issue by: 1. make ph_key for TrainModelInput to be model_input.id_list_features["defalut"] 2. handles parsing in trec utils. Differential Revision: D62160686
1 parent 07d794f commit 135657a

File tree

1 file changed

+2
-1
lines changed
  • torchrec/distributed/train_pipeline

1 file changed

+2
-1
lines changed

torchrec/distributed/train_pipeline/utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -774,7 +774,8 @@ def _get_node_args_helper(
774774
# pyre-ignore[16]
775775
ph_key: str = child_node.ph_key
776776
# example: ph_key = 'event_id_list_features_seqs[marketplace]'
777-
ph_keys = ph_key.split("[")
777+
ph_key = ph_key.replace("[", ".")
778+
ph_keys = ph_key.split(".")
778779
for key in ph_keys:
779780
if "]" in key:
780781
arg_info.input_attrs.append(key[:-1])

0 commit comments

Comments
 (0)