diff --git a/donkeycar/templates/complete.py b/donkeycar/templates/complete.py index 95770f7296..16dcd1e86a 100644 --- a/donkeycar/templates/complete.py +++ b/donkeycar/templates/complete.py @@ -64,10 +64,13 @@ def drive(cfg, model_path=None, use_joystick=False, model_type=None, #if we also try to use cuda. so disable for donkey_gym. os.environ["CUDA_VISIBLE_DEVICES"]="-1" + train_localizer = getattr(cfg, 'TRAIN_LOCALIZER', False) + train_behaviors = getattr(cfg, 'TRAIN_BEHAVIORS', False) + if model_type is None: - if cfg.TRAIN_LOCALIZER: + if train_localizer: model_type = "localizer" - elif cfg.TRAIN_BEHAVIORS: + elif train_behaviors: model_type = "behavior" else: model_type = cfg.DEFAULT_MODEL_TYPE @@ -394,7 +397,7 @@ def run(self, *components): # outputs = ['pilot/angle', 'pilot/throttle'] - if cfg.TRAIN_LOCALIZER: + if train_localizer: outputs.append("pilot/loc") # diff --git a/donkeycar/tests/test_template.py b/donkeycar/tests/test_template.py index 4daee0bb88..8b56bbfd6a 100755 --- a/donkeycar/tests/test_template.py +++ b/donkeycar/tests/test_template.py @@ -18,7 +18,7 @@ def test_config(): def test_drive(): - path = default_template(d2_path(gettempdir())) + path = custom_template(d2_path(gettempdir()), template="complete") myconfig = open(os.path.join(path, 'myconfig.py'), "wt") myconfig.write("CAMERA_TYPE = 'MOCK'\n") myconfig.write("USE_SSD1306_128_32 = False \n") diff --git a/pyproject.toml b/pyproject.toml index 72cb37dd3f..f6520485c8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -116,6 +116,9 @@ donkey = "donkeycar.management.base:execute_from_command_line" [project.urls] Homepage = "https://github.com/autorope/donkeycar" +[tool.pytest.ini_options] +norecursedirs = ["scripts"] + [tool.setuptools.dynamic] version = {attr = "donkeycar.__version__"}