diff --git a/axlearn/cloud/gcp/jobs/dataflow.py b/axlearn/cloud/gcp/jobs/dataflow.py index 054022d68..840492aa2 100644 --- a/axlearn/cloud/gcp/jobs/dataflow.py +++ b/axlearn/cloud/gcp/jobs/dataflow.py @@ -177,6 +177,7 @@ def from_flags(cls, fv: flags.FlagValues, **kwargs): sorted(flags.flag_dict_to_args(dataflow_spec, multi_flags=multi_flags)) ) cfg.setup_command = f"{docker_setup_cmd} && {docker_auth_cmd} && {bundle_cmd}" + cfg.command = cfg.command.strip("'\"") cfg.command = f"{cfg.command} {dataflow_flags}" return cfg