From 299ead1598e62173455b4ff5bed298a93a343bd5 Mon Sep 17 00:00:00 2001 From: Kiuk Chung Date: Fri, 28 Jan 2022 16:18:21 -0800 Subject: [PATCH] (torchx/cli) load .torchxconfig file from HOME -> CWD for cmd_run Summary: Adds `$HOME` to the list of dirs where we search for .torchxconfig. $HOME/.torchxconfig takes precedence over $CWD/.torchxconfig Reviewed By: d4l3k Differential Revision: D33858048 fbshipit-source-id: 6a3dc385bdbcb219f9f0dd83fa110c2636216f04 --- torchx/cli/cmd_run.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/torchx/cli/cmd_run.py b/torchx/cli/cmd_run.py index dbe524ac5..fa8f4aa2c 100644 --- a/torchx/cli/cmd_run.py +++ b/torchx/cli/cmd_run.py @@ -10,6 +10,7 @@ import sys import threading from dataclasses import asdict +from pathlib import Path from pprint import pformat from typing import Dict, List, Optional, Tuple, Type @@ -196,15 +197,19 @@ def _run(self, runner: Runner, args: argparse.Namespace) -> None: run_opts = runner.run_opts() scheduler_opts = run_opts[args.scheduler] cfg = _parse_run_config(args.scheduler_args, scheduler_opts) - config.apply(scheduler=args.scheduler, cfg=cfg) + config_dirs = [str(Path.home()), str(Path.cwd())] - config_files = config.find_configs() + config.apply(scheduler=args.scheduler, cfg=cfg, dirs=config_dirs) + + config_files = config.find_configs(dirs=config_dirs) workspace = ( "file://" + os.path.dirname(config_files[0]) if config_files else None ) + component, component_args = _parse_component_name_and_args( args.component_name_and_args, none_throws(self._subparser), + dirs=config_dirs, ) try: