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

aap_utils/__init__.py

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

aap_utils/register.py

@@ -63,18 +63,28 @@ def create_url(domain_name = "aiplatform.vcntt.tech"):
 
 def get_args():
     kcn_params = "config.yaml"
-    config_path="configs/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
+
     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.6.dist-info → aap_utils-0.1.8.dist-info}/METADATA

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

aap_utils-0.1.8.dist-info/RECORD

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

aap_utils/init_kcn.py

@@ -1,19 +0,0 @@
-import os
-from hydra import initialize, compose
-from omegaconf import OmegaConf, DictConfig
-kcn_params = "config.yaml"
-config_path="configs/"
-overrides = []
-
-if os.environ.get('KCN_PARAMS') and os.environ.get('KCN_PARAMS') != '':
-    full_path = os.getenv('KCN_PARAMS')
-    full_path = "../" + full_path
-    config_path = os.path.dirname(full_path)
-    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 []
-
-with initialize(version_base=None, config_path=config_path):
-    args = compose(config_name=kcn_params)

aap_utils-0.1.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-aap_utils/__init__.py,sha256=QcqBOrCuKWHsW7wNycAyPJTPquUzw_dg8E98XTm7OUk,67
-aap_utils/init_kcn.py,sha256=FMX9w0WoaP6I3yc3xn8xqGYT2VbtG2uJbwpTKouZpew,718
-aap_utils/register.py,sha256=YDUvUtoICYVMuuzYAueb34ZtGcYI8SJ6YxoNB3POyyQ,3069
-aap_utils-0.1.6.dist-info/METADATA,sha256=icRBc45ty6jVl1bXuZUQw52XCvMW7yWG8j21_OXkGA4,346
-aap_utils-0.1.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-aap_utils-0.1.6.dist-info/top_level.txt,sha256=Qt5xmL8_0_jhMl9uA3lg-xnvs4nCEyVj1tBE4vnp36M,10
-aap_utils-0.1.6.dist-info/RECORD,,