aiauto-client 0.1.21__py3-none-any.whl → 0.1.22__py3-none-any.whl

aiauto/core.py

@@ -1,4 +1,5 @@
 from os import makedirs
+import os
 import tempfile
 from typing import Union, Optional, List, Dict, Callable
 import optuna
@@ -44,9 +45,8 @@ class AIAutoController:
                 "Please delete and reissue your token from the web dashboard at https://dashboard.common.aiauto.pangyo.ainode.ai"
             ) from e
 
-        # artifact storage
-        makedirs('./artifacts', exist_ok=True)
-        self.artifact_store = optuna.artifacts.FileSystemArtifactStore('./artifacts')
+        # artifact storage: lazily resolved at call time based on runtime environment
+        self._artifact_store = None
         self.tmp_dir = tempfile.mkdtemp(prefix=f'ai_auto_tmp_')
 
     def get_storage(self):
@@ -57,7 +57,18 @@ class AIAutoController:
         optuna.artifacts.Boto3ArtifactStore,
         optuna.artifacts.GCSArtifactStore,
     ]:
-        return self.artifact_store
+        # Lazy init: prefer container-mounted PVC at /artifacts (runner env),
+        # fallback to local ./artifacts for client/local usage.
+        if self._artifact_store is None:
+            if os.path.isdir('/artifacts'):
+                path = '/artifacts'
+            else:
+                path = './artifacts'
+                if not os.path.isdir(path):
+                    makedirs(path, exist_ok=True)
+            self._artifact_store = optuna.artifacts.FileSystemArtifactStore(path)
+
+        return self._artifact_store
 
     def get_artifact_tmp_dir(self):
         return self.tmp_dir

{aiauto_client-0.1.21.dist-info → aiauto_client-0.1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.21
+Version: 0.1.22
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://dashboard.common.aiauto.pangyo.ainode.ai

{aiauto_client-0.1.21.dist-info → aiauto_client-0.1.22.dist-info}/RECORD

@@ -1,10 +1,10 @@
 aiauto/__init__.py,sha256=sF7sJaXg7-MqolSYLxsaXAir1dBzARhXLrHo7zLsupg,345
 aiauto/_config.py,sha256=hTFh2bH9m-HuX6QCpNtBC0j6rEB0S97hhPKjbEjv4Tg,89
 aiauto/constants.py,sha256=rBibGOQHHrdkwaai92-3I8-N0cu-B4CoCoQbG9-Cl8k,821
-aiauto/core.py,sha256=cHTkg-K7Q1WvGGlOZN4Vof0UzCra2jf6YFWUGTtQXuQ,10735
+aiauto/core.py,sha256=qQkLpeWIQZwXeYvbdUw9p2p--0vZx3rZLpHb4N7vj9o,11203
 aiauto/http_client.py,sha256=v_nPdb-2tIeH1XrOYqzMGvFfXLKEDbQoSaQYPsB0Hik,2587
 aiauto/serializer.py,sha256=BJmeq6uCD9D2_6bXu_sMBQLSsXCUMIMM10iX923DTXE,2749
-aiauto_client-0.1.21.dist-info/METADATA,sha256=utaLL9kxOHFigDto-2zdkDLlBONC51_olXfir9zhL3U,25746
-aiauto_client-0.1.21.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-aiauto_client-0.1.21.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
-aiauto_client-0.1.21.dist-info/RECORD,,
+aiauto_client-0.1.22.dist-info/METADATA,sha256=TprOc3A7bMm3tt61-gXvq8HWAok3vxoi9XoXlppGN90,25746
+aiauto_client-0.1.22.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+aiauto_client-0.1.22.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
+aiauto_client-0.1.22.dist-info/RECORD,,