aap-utils 0.1.8__py3-none-any.whl → 0.1.11__py3-none-any.whl

aap_utils/__init__.py

@@ -1 +1,2 @@
-from .register import register_demo, get_args
+from .register import register_demo
+from .config_loader import ConfigLoader

aap_utils/config_loader.py

@@ -0,0 +1,26 @@
+import os
+from hydra import initialize, compose
+from omegaconf import DictConfig
+
+class ConfigLoader:
+    def __init__(self):
+        self.kcn_params = "config.yaml"
+        self.config_path = "configs/"
+        self.overrides = []
+
+    def _load_env_variables(self):
+        """Load environment variables for config parameters and overrides."""
+        if os.environ.get("KCN_PARAMS"):
+            full_path = os.getenv("KCN_PARAMS")
+            self.kcn_params = os.path.basename(full_path)  # Extract the filename
+
+        if os.environ.get("KCN_OVERRIDES"):
+            overrides_env = os.getenv("KCN_OVERRIDES")
+            self.overrides = overrides_env.split("|") if overrides_env else []
+
+    def get_args(self) -> DictConfig:
+        """Load and return the Hydra configuration."""
+        self._load_env_variables()
+        with initialize(version_base=None, config_path=self.config_path):
+            args = compose(config_name=self.kcn_params, overrides=self.overrides)
+        return args

aap_utils/register.py

@@ -63,28 +63,18 @@ def create_url(domain_name = "aiplatform.vcntt.tech"):
 
 def get_args():
     kcn_params = "config.yaml"
-
-    # Get the directory of run.py (assuming run.py is executed from its own location)
-    run_py_path = os.getcwd()  # This gets the directory where run.py is executed
-
-    # Relative path to configs directory
-    config_path = "configs"  # Hydra requires a relative path
-
+    config_path="configs/"
     overrides = []
 
     if os.environ.get('KCN_PARAMS') and os.environ.get('KCN_PARAMS') != '':
         full_path = os.getenv('KCN_PARAMS')
+        config_path = os.path.join(config_path, kcn_params)
         kcn_params = os.path.basename(full_path)  # Extract the filename
 
     if os.environ.get('KCN_OVERRIDES') and os.environ.get('KCN_OVERRIDES') != '':
         overrides_env = os.getenv("KCN_OVERRIDES")
         overrides = overrides_env.split('|') if overrides_env else []
 
-    # Ensure the config directory exists
-    if not os.path.exists(os.path.join(run_py_path, config_path)):
-        raise FileNotFoundError(f"Config directory not found: {os.path.join(run_py_path, config_path)}")
-
     with initialize(version_base=None, config_path=config_path):
         args = compose(config_name=kcn_params)
-
     return args

{aap_utils-0.1.8.dist-info → aap_utils-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aap_utils
-Version: 0.1.8
+Version: 0.1.11
 Summary: A Python library for registering an IP address for AAP
 Author: Minh Dang
 Author-email: danghoangminh86@gmail.com

aap_utils-0.1.11.dist-info/RECORD

@@ -0,0 +1,7 @@
+aap_utils/__init__.py,sha256=MiN8qoJdlw-cI6OLAfz1JuTOaIb0uEwngJrzCes6Zcg,75
+aap_utils/config_loader.py,sha256=RfYrBVnli7Jd1zPonpOhMd8kJC6zhaMPIq8QYKxYSfw,1001
+aap_utils/register.py,sha256=JxMfbfzc8qs3Z0QlbX7PiLgJexY2P0Zdq6fqdly6zsI,3058
+aap_utils-0.1.11.dist-info/METADATA,sha256=8gZpkUMOfsR5-10ONYs0_v19fk3qenYmMFBVQEuC9TU,347
+aap_utils-0.1.11.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+aap_utils-0.1.11.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.1.11.dist-info/RECORD,,

aap_utils-0.1.8.dist-info/RECORD

@@ -1,6 +0,0 @@
-aap_utils/__init__.py,sha256=UB0P2cVBboRNY4SE__iGw0Nscwv9KJFTINuwydrWv4M,45
-aap_utils/register.py,sha256=-TGFMUjgv8aIZBMZ6h9VkYoLqXAkAoi5_VJZyhcJ6Ys,3460
-aap_utils-0.1.8.dist-info/METADATA,sha256=14ly6dglTdHhGDF6jXnpeFCeGlCB8Ucx8MH_RZSfXpA,346
-aap_utils-0.1.8.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-aap_utils-0.1.8.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
-aap_utils-0.1.8.dist-info/RECORD,,