aap-utils 0.1.7__tar.gz → 0.1.11__tar.gz

{aap_utils-0.1.7 → aap_utils-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aap_utils
-Version: 0.1.7
+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/aap_utils/__init__.py

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

aap_utils-0.1.11/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-0.1.7 → aap_utils-0.1.11}/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 → aap_utils-0.1.11}/aap_utils.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aap_utils
-Version: 0.1.7
+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.7 → aap_utils-0.1.11}/aap_utils.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 setup.py
 aap_utils/__init__.py
+aap_utils/config_loader.py
 aap_utils/register.py
 aap_utils.egg-info/PKG-INFO
 aap_utils.egg-info/SOURCES.txt

{aap_utils-0.1.7 → aap_utils-0.1.11}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aap_utils",  # The name of your library
-    version="0.1.7",
+    version="0.1.11",
     description="A Python library for registering an IP address for AAP",
     author="Minh Dang",
     author_email="danghoangminh86@gmail.com",

aap_utils-0.1.7/aap_utils/__init__.py

@@ -1 +0,0 @@
-from .register import register_demo, get_args