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

aap_utils/__init__.py

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

aap_utils/config_loader.py

@@ -21,6 +21,6 @@ class ConfigLoader:
     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):
+        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/register.py

@@ -60,21 +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"
-    config_path="configs/"
+    
+    # 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')
-        config_path = os.path.join(config_path, 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 []
 
-    with initialize(version_base=None, config_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_config_dir(version_base=None, config_dir=config_path):
         args = compose(config_name=kcn_params)
-    return args
+
+    return args

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aap_utils
-Version: 0.1.11
+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.dist-info/RECORD

@@ -0,0 +1,7 @@
+aap_utils/__init__.py,sha256=iN7GQizkWVefysHlKpDG6L35Dkfaur2x2uS2uXtUiN4,46
+aap_utils/config_loader.py,sha256=dhwQYuJ7y1exe6zI8zRK8WL6XhXnlb9eDLNKG3GgNNs,1011
+aap_utils/register.py,sha256=GHzj4HSZ0ylF4UEADMRwDOq6sWdsKLr7QRBF7p6QYbs,3396
+aap_utils-0.1.12.dist-info/METADATA,sha256=DreyF-hB1jwyn7Qv4c8AiB79kOJeb17kQFs6YbcEMao,347
+aap_utils-0.1.12.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+aap_utils-0.1.12.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
+aap_utils-0.1.12.dist-info/RECORD,,

aap_utils-0.1.11.dist-info/RECORD

@@ -1,7 +0,0 @@
-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,,