aap-utils 0.1.7__py3-none-any.whl → 0.1.10__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,9 +63,7 @@ def create_url(domain_name = "aiplatform.vcntt.tech"):
 
 def get_args():
     kcn_params = "config.yaml"
-    
-    run_py_path = os.path.abspath(os.path.dirname(__file__))  # Get the directory of this script
-    config_path = os.path.join(run_py_path, "configs")  # Set config_path relative to run.py
+    config_path="configs/"
     overrides = []
 
     if os.environ.get('KCN_PARAMS') and os.environ.get('KCN_PARAMS') != '':

{aap_utils-0.1.7.dist-info → aap_utils-0.1.10.dist-info}/METADATA

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

aap_utils-0.1.10.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.10.dist-info/METADATA,sha256=EL3HO1NzrueCf5ajoqo6nZQaIHINo1jdnpfsSzADbbM,347
+aap_utils-0.1.10.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+aap_utils-0.1.10.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.1.10.dist-info/RECORD,,

aap_utils-0.1.7.dist-info/RECORD

@@ -1,6 +0,0 @@
-aap_utils/__init__.py,sha256=UB0P2cVBboRNY4SE__iGw0Nscwv9KJFTINuwydrWv4M,45
-aap_utils/register.py,sha256=LhyPdo375TAR8rb-adN9PJaTsDrd10StfX1QX_YRblw,3226
-aap_utils-0.1.7.dist-info/METADATA,sha256=-l64FenVgq64Fzh-TmXXxtxnoMEWJL7Co-yiuu9ulSE,346
-aap_utils-0.1.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-aap_utils-0.1.7.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
-aap_utils-0.1.7.dist-info/RECORD,,