aiauto-client 0.1.25__tar.gz → 0.1.27__tar.gz

{aiauto_client-0.1.25/src/aiauto_client.egg-info → aiauto_client-0.1.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.25
+Version: 0.1.27
 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.25 → aiauto_client-0.1.27}/pyproject.toml

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

{aiauto_client-0.1.25 → aiauto_client-0.1.27}/src/aiauto/core.py

@@ -162,34 +162,14 @@ 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_id를 user_attrs에서 완전히 제거
-                # GrpcStorageProxy는 빈 문자열을 제대로 처리하지 못하므로
-                # 전체 user_attrs를 다시 설정하는 방식 사용
-                try:
-                    # 현재 user_attrs 복사
-                    new_attrs = dict(old_trial.user_attrs)
-                    # artifact_id 키 제거
-                    if self.check_attr_name in new_attrs:
-                        del new_attrs[self.check_attr_name]
-                    
-                    # 각 속성을 다시 설정 (artifact_id 제외)
-                    for key, value in new_attrs.items():
-                        study._storage.set_trial_user_attr(old_trial._trial_id, key, value)
-                    
-                    # artifact_id를 명시적으로 None으로 설정 시도
-                    # (일부 storage는 None을 "키 삭제"로 해석할 수 있음)
-                    try:
-                        study._storage.set_trial_user_attr(old_trial._trial_id, self.check_attr_name, None)
-                    except:
-                        pass  # None이 지원되지 않으면 무시
-                        
-                except Exception as e2:
-                    print(f"Warning: Failed to remove user_attr {self.check_attr_name} for trial {old_trial.number}: {e2}")
+                # artifact_removed 플래그 추가 (WAL 방식의 storage에서도 안전함)
+                study._storage.set_trial_user_attr(old_trial._trial_id, 'artifact_removed', True)
 
 
 class StudyWrapper:

{aiauto_client-0.1.25 → aiauto_client-0.1.27/src/aiauto_client.egg-info}/PKG-INFO

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