Commit f8d0c32b authored by nilabha's avatar nilabha

added obs_utils for converting demonstration, train.py code made generic to...

added obs_utils for converting demonstration, train.py code made generic to allow apex mixed IL and RL
parent f582ff19
Pipeline #4955 passed with stage
in 30 minutes and 20 seconds
......@@ -98,12 +98,13 @@ def run(args, parser):
# Bazel makes it hard to find files specified in `args` (and `data`).
# Look for them here.
# NOTE: Some of our yaml files don't have a `config` section.
if exp.get("config", {}).get("input") and \
not os.path.exists(exp["config"]["input"]):
# This script runs in the ray/rllib dir.
rllib_dir = Path(__file__).parent
input_file = rllib_dir.absolute().joinpath(exp["config"]["input"])
exp["config"]["input"] = str(input_file)
if exp.get("config", {}).get("input"):
if not isinstance(exp.get("config", {}).get("input"),dict):
if not os.path.exists(exp["config"]["input"]):
# This script runs in the ray/rllib dir.
rllib_dir = Path(__file__).parent
input_file = rllib_dir.absolute().joinpath(exp["config"]["input"])
exp["config"]["input"] = str(input_file)
if not exp.get("run"):
parser.error("the following arguments are required: --run")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment