aiauto-client 0.1.21__tar.gz → 0.1.23__tar.gz

{aiauto_client-0.1.21/src/aiauto_client.egg-info → aiauto_client-0.1.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.21
+Version: 0.1.23
 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 → aiauto_client-0.1.23}/examples/example_torch_multiple_objective.py

@@ -226,7 +226,7 @@ def objective_multi(trial):
 
     # ====================== 학습 완료 후 최종 모델 스냅샷 저장 ========================
     try:
-        filename = f'{trial.study.study_name}_{trial.number}'
+        filename = f'{trial.study.study_name}_{trial.number}.pt'
         model_path = join(ac.get_artifact_tmp_dir(), filename)
         # 모델 임시 폴더에 저장
         torch.save(
@@ -396,7 +396,7 @@ def objective_detailed(trial):
 
     # ====================== 학습 완료 후 최종 모델 스냅샷 저장 ========================
     try:
-        filename = f'{trial.study.study_name}_{trial.number}'
+        filename = f'{trial.study.study_name}_{trial.number}.pt'
         model_path = join(ac.get_artifact_tmp_dir(), filename)
         # 모델 임시 폴더에 저장
         torch.save(

{aiauto_client-0.1.21 → aiauto_client-0.1.23}/examples/example_torch_single_objective.py

@@ -230,7 +230,7 @@ def objective_single(trial):
 
     # ====================== 학습 완료 후 최종 모델 스냅샷 저장 ========================
     try:
-        filename = f'{trial.study.study_name}_{trial.number}'
+        filename = f'{trial.study.study_name}_{trial.number}.pt'
         model_path = join(ac.get_artifact_tmp_dir(), filename)
         # 모델 임시 폴더에 저장
         torch.save(
@@ -405,7 +405,7 @@ def objective_detailed(trial):
 
     # ====================== 학습 완료 후 최종 모델 스냅샷 저장 ========================
     try:
-        filename = f'{trial.study.study_name}_{trial.number}'
+        filename = f'{trial.study.study_name}_{trial.number}.pt'
         model_path = join(ac.get_artifact_tmp_dir(), filename)
         # 모델 임시 폴더에 저장
         torch.save(

{aiauto_client-0.1.21 → aiauto_client-0.1.23}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aiauto-client"
-version = "0.1.21"
+version = "0.1.23"
 description = "AI Auto HPO (Hyperparameter Optimization) Client Library"
 readme = "README.md"
 requires-python = ">=3.8"

{aiauto_client-0.1.21 → aiauto_client-0.1.23}/src/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 → aiauto_client-0.1.23/src/aiauto_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.21
+Version: 0.1.23
 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