aiauto-client 0.1.7__tar.gz → 0.1.9__tar.gz

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.7
+Version: 0.1.9
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://dashboard.aiauto.pangyo.ainode.ai
@@ -94,6 +94,17 @@ study_wrapper.optimize(
     runtime_image=aiauto.RUNTIME_IMAGES[0],
 )
 ```
+- 종료 됐는지 optuna-dashboard 가 아닌 코드로 확인하는 법
+```python
+study_wrapper.get_status()
+# {'study_name': 'test', 'count_active': 0, 'count_succeeded': 10, 'count_pruned': 0, 'count_failed': 0, 'count_total': 10, 'count_completed': 10, 'dashboard_url': 'https://optuna-dashboard-10f804bb-52be-48e8-aa06-9f5411ed4b0d.aiauto.pangyo.ainode.ai', 'last_error': '', 'updated_at': '2025-09-01T11:31:49.375Z'}
+while study_wrapper.get_status()['count_completed'] <= study_wrapper.get_status()['count_total']:
+    sleep(10)  # 10초 마다 한 번 씩
+```
+- best trial 을 가져오는 법
+```python
+TODO
+```
 
 ## Jupyter Notebook 사용 시 주의사항
 

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/README.md

@@ -67,6 +67,17 @@ study_wrapper.optimize(
     runtime_image=aiauto.RUNTIME_IMAGES[0],
 )
 ```
+- 종료 됐는지 optuna-dashboard 가 아닌 코드로 확인하는 법
+```python
+study_wrapper.get_status()
+# {'study_name': 'test', 'count_active': 0, 'count_succeeded': 10, 'count_pruned': 0, 'count_failed': 0, 'count_total': 10, 'count_completed': 10, 'dashboard_url': 'https://optuna-dashboard-10f804bb-52be-48e8-aa06-9f5411ed4b0d.aiauto.pangyo.ainode.ai', 'last_error': '', 'updated_at': '2025-09-01T11:31:49.375Z'}
+while study_wrapper.get_status()['count_completed'] <= study_wrapper.get_status()['count_total']:
+    sleep(10)  # 10초 마다 한 번 씩
+```
+- best trial 을 가져오는 법
+```python
+TODO
+```
 
 ## Jupyter Notebook 사용 시 주의사항
 

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/examples/example_torch_multiple_objective.py

@@ -27,7 +27,7 @@ ac = aiauto.AIAutoController('<token>')
 
 
 # multi objective without pruning (accuracy + FLOPS)
-def objective_multi(trial: optuna.trial.Trial):
+def objective_multi(trial):
     """
     다중 목적 함수: accuracy 최대화 + FLOPS 최소화
     hyper parameter tuning을 할 때는 데이터의 일부만 사용하여 빠르게 HPO를 한다

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/examples/example_torch_single_objective.py

@@ -27,7 +27,7 @@ ac = aiauto.AIAutoController('<token>')
 
 
 # single objective accuracy with pruning
-def objective_single(trial: optuna.trial.Trial):
+def objective_single(trial):
     """
     단일 목적 함수: accuracy 최대화
     hyper parameter tuning 할 때는 데이터의 일부만 사용하여 빠르게 HPO를 한다

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/examples/simple_example.py

@@ -11,7 +11,6 @@
   $ python simple_example.py
 """
 
-import optuna
 import aiauto
 
 
@@ -26,7 +25,9 @@ ac = aiauto.AIAutoController('<token>')
 
 
 # single objective accuracy
-def simple_objective(trial: optuna.trial.Trial):
+def simple_objective(trial):
+    import optuna
+    import aiauto
     """
     간단한 2차 함수 최적화
     (x - 2)^2 + (y - 3)^2 를 최소화

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/pyproject.toml

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

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/src/aiauto/core.py

@@ -31,8 +31,16 @@ class AIAutoController:
             if not host_external:
                 raise RuntimeError("No storage host returned from EnsureWorkspace")
 
-            host, port = host_external.split(':')
-            self.storage = optuna.storages.GrpcStorageProxy(host=host, port=int(port))
+            # Parse host and port
+            if ':' in host_external:
+                host, port_str = host_external.rsplit(':', 1)
+                port = int(port_str)
+            else:
+                host = host_external
+                port = 443  # Default to HTTPS port
+            
+            # Create storage with TLS support for port 443
+            self.storage = self._create_storage_with_tls(host, port)
 
             # Store the internal host for CRD usage (if needed later)
             self.storage_host_internal = response.get('journalGrpcStorageProxyHostInternal', '')
@@ -49,6 +57,74 @@ class AIAutoController:
         self.artifact_store = optuna.artifacts.FileSystemArtifactStore('./artifacts')
         self.tmp_dir = tempfile.mkdtemp(prefix=f'ai_auto_tmp_')
 
+    def _create_storage_with_tls(self, host: str, port: int):
+        """Create GrpcStorageProxy with automatic TLS detection based on port"""
+        # Port 13000 = internal (plain), Port 443 = external (TLS)
+        if port != 443:
+            # Plain gRPC for internal connections
+            return optuna.storages.GrpcStorageProxy(host=host, port=port)
+        
+        # TLS connection for external access (port 443)
+        import grpc
+        creds = grpc.ssl_channel_credentials()
+        
+        # Try different TLS parameter names for Optuna version compatibility
+        # Try 1: channel_credentials parameter (newer Optuna versions)
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host, 
+                port=port, 
+                channel_credentials=creds
+            )
+        except TypeError:
+            pass
+        
+        # Try 2: ssl boolean parameter
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host, 
+                port=port, 
+                ssl=True
+            )
+        except TypeError:
+            pass
+        
+        # Try 3: use_tls parameter
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host,
+                port=port,
+                use_tls=True
+            )
+        except TypeError:
+            pass
+        
+        # Try 4: secure parameter
+        try:
+            return optuna.storages.GrpcStorageProxy(
+                host=host,
+                port=port,
+                secure=True
+            )
+        except TypeError:
+            pass
+        
+        # Fallback: try to create secure channel manually
+        try:
+            channel = grpc.secure_channel(f"{host}:{port}", creds)
+            # Some Optuna versions might accept a channel directly
+            return optuna.storages.GrpcStorageProxy(channel=channel)
+        except (TypeError, AttributeError):
+            pass
+        
+        # If all attempts fail, raise informative error
+        raise RuntimeError(
+            f"Failed to create TLS connection to {host}:{port}. "
+            "GrpcStorageProxy TLS parameters not recognized. "
+            "Please check Optuna version compatibility. "
+            "Tried: channel_credentials, ssl, use_tls, secure, channel parameters."
+        )
+    
     def get_storage(self):
         return self.storage
 
@@ -172,7 +248,7 @@ class StudyWrapper:
                 self._study = optuna.create_study(
                     study_name=self.study_name,
                     storage=self._storage,
-                    load_if_exists=True
+                    load_if_exists=True,
                 )
             except Exception as e:
                 raise RuntimeError(
@@ -202,7 +278,8 @@ class StudyWrapper:
             request_data = {
                 "objective": {
                     "sourceCode": serialize(objective),
-                    "requirementsTxt": build_requirements(requirements_file, requirements_list)
+                    "requirementsTxt": build_requirements(requirements_file, requirements_list),
+                    "objectiveName": objective.__name__  # 함수명 추출하여 전달
                 },
                 "batch": {
                     "studyName": self.study_name,

{aiauto_client-0.1.7 → aiauto_client-0.1.9}/src/aiauto_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.7
+Version: 0.1.9
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://dashboard.aiauto.pangyo.ainode.ai
@@ -94,6 +94,17 @@ study_wrapper.optimize(
     runtime_image=aiauto.RUNTIME_IMAGES[0],
 )
 ```
+- 종료 됐는지 optuna-dashboard 가 아닌 코드로 확인하는 법
+```python
+study_wrapper.get_status()
+# {'study_name': 'test', 'count_active': 0, 'count_succeeded': 10, 'count_pruned': 0, 'count_failed': 0, 'count_total': 10, 'count_completed': 10, 'dashboard_url': 'https://optuna-dashboard-10f804bb-52be-48e8-aa06-9f5411ed4b0d.aiauto.pangyo.ainode.ai', 'last_error': '', 'updated_at': '2025-09-01T11:31:49.375Z'}
+while study_wrapper.get_status()['count_completed'] <= study_wrapper.get_status()['count_total']:
+    sleep(10)  # 10초 마다 한 번 씩
+```
+- best trial 을 가져오는 법
+```python
+TODO
+```
 
 ## Jupyter Notebook 사용 시 주의사항