aap-utils 0.1.8__tar.gz → 0.1.12__tar.gz

{aap_utils-0.1.8 → aap_utils-0.1.12}/PKG-INFO

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

aap_utils-0.1.12/aap_utils/__init__.py

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

aap_utils-0.1.12/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_config_dir(version_base=None, config_dir=self.config_path):
+            args = compose(config_name=self.kcn_params, overrides=self.overrides)
+        return args

{aap_utils-0.1.8 → aap_utils-0.1.12}/aap_utils/register.py

@@ -60,31 +60,28 @@ def create_url(domain_name = "aiplatform.vcntt.tech"):
     url = f"https://{transformed_namespace}.{domain_name}"
     return url, transformed_namespace
 
-
 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
+    
+    # Get the absolute path of the folder where run.py is located
+    run_py_dir = Path(__file__).parent.resolve()
+    config_path = os.path.join(run_py_dir, "configs")  # Absolute path to "configs/"
 
     overrides = []
 
-    if os.environ.get('KCN_PARAMS') and os.environ.get('KCN_PARAMS') != '':
-        full_path = os.getenv('KCN_PARAMS')
+    if os.environ.get("KCN_PARAMS") and os.environ.get("KCN_PARAMS") != "":
+        full_path = os.getenv("KCN_PARAMS")
         kcn_params = os.path.basename(full_path)  # Extract the filename
 
-    if os.environ.get('KCN_OVERRIDES') and os.environ.get('KCN_OVERRIDES') != '':
+    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 []
+        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)}")
+    # Ensure the configs directory exists
+    if not os.path.exists(os.path.join(config_path, kcn_params)):
+        raise FileNotFoundError(f"Config file not found: {os.path.join(config_path, kcn_params)}")
 
-    with initialize(version_base=None, config_path=config_path):
+    with initialize_config_dir(version_base=None, config_dir=config_path):
         args = compose(config_name=kcn_params)
 
-    return args
+    return args

{aap_utils-0.1.8 → aap_utils-0.1.12}/aap_utils.egg-info/PKG-INFO

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

{aap_utils-0.1.8 → aap_utils-0.1.12}/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.8 → aap_utils-0.1.12}/setup.py

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

aap_utils-0.1.8/aap_utils/__init__.py

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