aiauto-client 0.1.23__py3-none-any.whl → 0.1.25__py3-none-any.whl

aiauto/core.py

@@ -166,13 +166,30 @@ class CallbackTopNArtifact:
                     self.artifact_store.remove(artifact_id)
                 except Exception as e:
                     print(f"Warning: Failed to remove artifact {artifact_id}: {e}")
-                finally:
+                
+                # 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:
-                        # 일부 스토리지(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}")
+                        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}")
 
 
 class StudyWrapper:

{aiauto_client-0.1.23.dist-info → aiauto_client-0.1.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.23
+Version: 0.1.25
 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.23.dist-info → aiauto_client-0.1.25.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=qQkLpeWIQZwXeYvbdUw9p2p--0vZx3rZLpHb4N7vj9o,11203
+aiauto/core.py,sha256=lVE3H3W7knnFbdwC2mRyL42TyEDyHK02crFaIuhkjgA,12007
 aiauto/http_client.py,sha256=v_nPdb-2tIeH1XrOYqzMGvFfXLKEDbQoSaQYPsB0Hik,2587
 aiauto/serializer.py,sha256=BJmeq6uCD9D2_6bXu_sMBQLSsXCUMIMM10iX923DTXE,2749
-aiauto_client-0.1.23.dist-info/METADATA,sha256=Gn6_mp6YHjc3P-UTHv3kQsgFXoG0KZHPxTWJgrRxGig,25746
-aiauto_client-0.1.23.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-aiauto_client-0.1.23.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
-aiauto_client-0.1.23.dist-info/RECORD,,
+aiauto_client-0.1.25.dist-info/METADATA,sha256=w0nyFqlPNrFXW7EIKZnEYPi6IWmEHdUcyq-S4c8kI90,25746
+aiauto_client-0.1.25.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+aiauto_client-0.1.25.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
+aiauto_client-0.1.25.dist-info/RECORD,,