pathlib.Path support in Config.get_config (#357)

Author: kozlov721

luxonis_ml/utils/config.py

@@ -1,5 +1,5 @@
 import ast
-from pathlib import PurePath
+from pathlib import Path, PurePath
 from typing import Any, TypeVar
 
 import yaml
@@ -19,7 +19,7 @@ class LuxonisConfig(BaseModelExtraForbid):
     @classmethod
     def get_config(
         cls,
-        cfg: str | Params | None = None,
+        cfg: PathType | Params | None = None,
         overrides: Params | list[str] | tuple[str, ...] | None = None,
     ) -> Self:
         """Loads config from a yaml file or a dictionary.
@@ -50,7 +50,9 @@ def get_config(
         overrides = overrides or {}
         cfg = cfg or {}
 
-        if isinstance(cfg, str):
+        if isinstance(cfg, Path):
+            data = yaml.safe_load(cfg.read_text(encoding="utf-8"))
+        elif isinstance(cfg, str):
             fs = LuxonisFileSystem(cfg)
             buffer = fs.read_to_byte_buffer()
             data = yaml.safe_load(buffer)

tests/test_utils/test_config.py

@@ -82,7 +82,7 @@ def test_invalid_config_path():
 
 
 def test_config_simple(config_file: str):
-    cfg = Config.get_config(config_file)
+    cfg = Config.get_config(Path(config_file))
     assert (
         cfg.sub_config.str_sub_param
         == CONFIG_DATA["sub_config"]["str_sub_param"]