From 5007ab2932402853ee4cd036d40cb9074dfaf82b Mon Sep 17 00:00:00 2001 From: bdvllrs Date: Wed, 8 Nov 2023 09:35:47 +0000 Subject: [PATCH] Add argv param to the load_structured_config func --- shimmer/config.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/shimmer/config.py b/shimmer/config.py index e3d4686b..4afe7058 100644 --- a/shimmer/config.py +++ b/shimmer/config.py @@ -24,7 +24,7 @@ def load_config( load_dirs: list[str] | None = None, use_cli: bool = True, debug_mode: bool = False, - argv: Any = None, + argv: list[str] | None = None, ) -> Config: config_path = Path(path) if not config_path.is_dir(): @@ -88,6 +88,7 @@ def load_structured_config( load_dirs: list[str] | None = None, use_cli: bool = True, debug_mode: bool = False, + argv: list[str] | None = None, ) -> T: - config = load_config(path, structure, load_dirs, use_cli, debug_mode) + config = load_config(path, structure, load_dirs, use_cli, debug_mode, argv) return cast(T, config)