Update jinja2 and error messages

project-lighter · Jun 6, 2024 · d983a60 · d983a60
1 parent 8e73be8
commit d983a60
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/lighter/utils/runner.py b/lighter/utils/runner.py
@@ -32,7 +32,7 @@ def parse_config(**kwargs) -> ConfigParser:
     # Ensure a config file is specified.
     config = kwargs.pop("config", None)
     if config is None:
-        raise ValueError("'--config' not specified. Exiting.")
+        raise ValueError("'--config' not specified. Please provide a valid configuration file.")
 
     # Read the config file and update it with overrides.
     parser = ConfigParser(CONFIG_STRUCTURE, globals=False)
@@ -60,7 +60,7 @@ def validate_config(parser: ConfigParser) -> None:
     root_keys = parser.get().keys()
     invalid_keys = set(root_keys) - set(CONFIG_STRUCTURE.keys()) - {"_meta_", "_requires_"}
     if invalid_keys:
-        raise ValueError(f"Invalid top-level config keys: {list(invalid_keys)}. Allowed keys: {list(CONFIG_STRUCTURE.keys())}")
+        raise ValueError(f"Invalid top-level config keys: {invalid_keys}. Allowed keys: {CONFIG_STRUCTURE.keys()}")
 
     # Validate that 'args' contains only valid Trainer/Tuner method names.
     args_keys = parser.get("args", {}).keys()

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -53,6 +53,7 @@ requests = "^2.31.0"
 monai = "^1.3.0"
 rich = "^13.7.0"
 torchvision = "^0.17.0"
+jinja2 = "^3.1.4"
 
 [tool.poetry.group.safety]
 optional = true