config.py improvements based on code from dfint/search_offsets

dfint64_patch/config.py

@@ -8,20 +8,26 @@
 T = TypeVar("T", bound=DictConfig)
 
 
-def with_config(config_class: type[T], config_file: str) -> Callable[[Callable[[T], None]], Callable[[], None]]:
+def with_config(config_class: type[T], *config_files: str) -> Callable[[Callable[[T], None]], Callable[[], None]]:
     """
     A decorator to load the config file and merge it with the CLI options.
+
+    :param config_class: A class, which describes the possible configuration parameters.
+        Should be inherited from `omegaconf.DictConfig`.
+    :param config_files: Names of configs. The closer a file is to the end of the list, the higher its priority.
+        Default config should be the first in the list. CLI options have the highest priority.
     """
 
     def decorator(func: Callable[[T], None]) -> Callable[[], None]:
         @functools.wraps(func)
         def wrapper() -> None:
             config = OmegaConf.structured(config_class)
 
-            try:
-                config.merge_with(OmegaConf.load(config_file))
-            except FileNotFoundError:
-                logger.info(f"Config {config_file} not found, using CLI options only")
+            for config_file in config_files:
+                try:
+                    config.merge_with(OmegaConf.load(config_file))
+                except FileNotFoundError:  # noqa: PERF203
+                    logger.info(f"Config {config_file} not found")
 
             config.merge_with(OmegaConf.from_cli())