aiauto-client 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

aiauto/__init__.py

@@ -2,13 +2,13 @@ from .core import AIAutoController, TrialController, CallbackTopNArtifact, Study
 from ._config import AIAUTO_API_TARGET
 from .constants import RUNTIME_IMAGES
 
-__version__ = "0.1.0"
+__version__ = "0.1.29"
 
 __all__ = [
     'AIAutoController',
-    'TrialController', 
+    'TrialController',
     'CallbackTopNArtifact',
     'StudyWrapper',
     'AIAUTO_API_TARGET',
     'RUNTIME_IMAGES',
-]
+]

aiauto/core.py

@@ -162,14 +162,17 @@ class CallbackTopNArtifact:
         for old_trial in finished_with_artifacts[self.n_keep:]:
             artifact_id = old_trial.user_attrs.get(self.check_attr_name)
             if artifact_id:
-                # 파일 삭제
                 try:
                     self.artifact_store.remove(artifact_id)
                 except Exception as e:
                     print(f"Warning: Failed to remove artifact {artifact_id}: {e}")
-                
-                # artifact_removed 플래그 추가 (WAL 방식의 storage에서도 안전함)
-                study._storage.set_trial_user_attr(old_trial._trial_id, 'artifact_removed', True)
+                finally:
+                    try:
+                        # 일부 스토리지(GrpcStorageProxy)는 None 값을 무시하거나 직렬화하지 않을 수 있음.
+                        # UI에서 링크를 숨기기 위해 빈 문자열로 설정(빈 문자열은 falsy로 처리됨).
+                        study._storage.set_trial_user_attr(old_trial._trial_id, self.check_attr_name, "")
+                    except Exception as e2:
+                        print(f"Warning: Failed to clear user_attr {self.check_attr_name} for trial {old_trial.number}: {e2}")
 
 
 class StudyWrapper:
@@ -214,10 +217,7 @@ class StudyWrapper:
                 resources_requests = {"cpu": "1", "memory": "1Gi"}
 
         if resources_limits is None:
-            if use_gpu:
-                resources_limits = {"cpu": "2", "memory": "4Gi"}
-            else:
-                resources_limits = {"cpu": "1", "memory": "1Gi"}
+            resources_limits = {}  # 빈 dict로 전달, operator가 requests 기반으로 처리
 
         if runtime_image is None or runtime_image == "":
             if use_gpu:

{aiauto_client-0.1.27.dist-info → aiauto_client-0.1.29.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.27
+Version: 0.1.29
 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.29.dist-info/RECORD

@@ -0,0 +1,10 @@
+aiauto/__init__.py,sha256=1O7vEwjbHS7lb6T686dg7RgBmaWHuGBvmm2kJHNRePE,346
+aiauto/_config.py,sha256=hTFh2bH9m-HuX6QCpNtBC0j6rEB0S97hhPKjbEjv4Tg,89
+aiauto/constants.py,sha256=rBibGOQHHrdkwaai92-3I8-N0cu-B4CoCoQbG9-Cl8k,821
+aiauto/core.py,sha256=5co4_Iyir5xuGInY5Rp1hav7hhUaHuoQiasoJ4Vj0Hw,11129
+aiauto/http_client.py,sha256=v_nPdb-2tIeH1XrOYqzMGvFfXLKEDbQoSaQYPsB0Hik,2587
+aiauto/serializer.py,sha256=BJmeq6uCD9D2_6bXu_sMBQLSsXCUMIMM10iX923DTXE,2749
+aiauto_client-0.1.29.dist-info/METADATA,sha256=fCWSx3oM0MfapSVoB1LDcMC4fnXBeU1Awbno-PpP4lk,25746
+aiauto_client-0.1.29.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+aiauto_client-0.1.29.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
+aiauto_client-0.1.29.dist-info/RECORD,,

aiauto_client-0.1.27.dist-info/RECORD

@@ -1,10 +0,0 @@
-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=n5nJsvzsnnBgBtdyecHcYxSian4NbD4IBl-cmdDag-w,10864
-aiauto/http_client.py,sha256=v_nPdb-2tIeH1XrOYqzMGvFfXLKEDbQoSaQYPsB0Hik,2587
-aiauto/serializer.py,sha256=BJmeq6uCD9D2_6bXu_sMBQLSsXCUMIMM10iX923DTXE,2749
-aiauto_client-0.1.27.dist-info/METADATA,sha256=tcd9TScT1WiEovn73ZKL8Vx0fqpzGrapbJeUnavbaNQ,25746
-aiauto_client-0.1.27.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-aiauto_client-0.1.27.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
-aiauto_client-0.1.27.dist-info/RECORD,,