aiauto-client 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

aiauto/__init__.py

@@ -1,5 +1,4 @@
 from .core import AIAutoController, TrialController, CallbackTopNArtifact, StudyWrapper
-from .api import create_study
 from ._config import AIAUTO_API_TARGET
 from .constants import RUNTIME_IMAGES
 
@@ -10,7 +9,6 @@ __all__ = [
     'TrialController', 
     'CallbackTopNArtifact',
     'StudyWrapper',
-    'create_study',
     'AIAUTO_API_TARGET',
     'RUNTIME_IMAGES',
 ]

aiauto/core.py

@@ -3,7 +3,7 @@ import tempfile
 from typing import Union, Optional, List, Dict, Callable
 import optuna
 from .http_client import ConnectRPCClient
-from .serializer import serialize, build_requirements
+from .serializer import serialize, build_requirements, object_to_json
 from ._config import AIAUTO_API_TARGET
 
 
@@ -62,6 +62,46 @@ class AIAutoController:
     def get_artifact_tmp_dir(self):
         return self.tmp_dir
 
+    def create_study(
+        self,
+        study_name: str,
+        direction: Optional[str] = None,
+        directions: Optional[List[str]] = None,
+        sampler: Union[object, dict, None] = None,
+        pruner: Union[object, dict, None] = None
+    ) -> 'StudyWrapper':
+        """Create a new study using the controller's token."""
+        if not direction and not directions:
+            raise ValueError("Either 'direction' or 'directions' must be specified")
+
+        if direction and directions:
+            raise ValueError("Cannot specify both 'direction' and 'directions'")
+
+        try:
+            # Prepare request data for CreateStudy
+            request_data = {
+                "spec": {
+                    "studyName": study_name,
+                    "direction": direction or "",
+                    "directions": directions or [],
+                    "samplerJson": object_to_json(sampler),
+                    "prunerJson": object_to_json(pruner)
+                }
+            }
+            
+            # Call CreateStudy RPC
+            response = self.client.call_rpc("CreateStudy", request_data)
+            
+            # Return StudyWrapper
+            return StudyWrapper(
+                study_name=response.get("studyName", study_name),
+                storage=self.storage,
+                controller=self
+            )
+
+        except Exception as e:
+            raise RuntimeError(f"Failed to create study: {e}") from e
+
 
 class TrialController:
     def __init__(self, trial: optuna.trial.Trial):

{aiauto_client-0.1.4.dist-info → aiauto_client-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.4
+Version: 0.1.5
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://aiauto.cloude.ainode.ai

aiauto_client-0.1.5.dist-info/RECORD

@@ -0,0 +1,10 @@
+aiauto/__init__.py,sha256=sF7sJaXg7-MqolSYLxsaXAir1dBzARhXLrHo7zLsupg,345
+aiauto/_config.py,sha256=DaRTIZlph9T3iuW-Cy4fkw8i3bXB--gMtW947SLZZNs,159
+aiauto/constants.py,sha256=UhDCLFoPE89XrHB3SEnZR3YUuzajgugMGX80KYx_qc0,939
+aiauto/core.py,sha256=BFKEF2wp3mjMFt5V7oGyZ531j_MIQBJNEiiOVhrKmzc,9549
+aiauto/http_client.py,sha256=t1gxeM5-d5bsVoFWgaNcTrt_WWUXuMuxge9gDlEqhoA,2086
+aiauto/serializer.py,sha256=_iPtEoqW8RTKOZ6UrC7CzOqoangpPYzeL7MQfIdmov8,1568
+aiauto_client-0.1.5.dist-info/METADATA,sha256=_tY0xtaoaz-ZimyBJSavhmApFa8p5oT9WuKDyHH7Hi0,3001
+aiauto_client-0.1.5.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+aiauto_client-0.1.5.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
+aiauto_client-0.1.5.dist-info/RECORD,,

aiauto/api.py

@@ -1,46 +0,0 @@
-from typing import Optional, List, Union
-from .serializer import object_to_json
-from .core import StudyWrapper, AIAutoController
-
-
-def create_study(
-    study_name: str,
-    token: str,
-    direction: Optional[str] = None,
-    directions: Optional[List[str]] = None,
-    sampler: Union[object, dict, None] = None,
-    pruner: Union[object, dict, None] = None
-) -> StudyWrapper:
-    if not direction and not directions:
-        raise ValueError("Either 'direction' or 'directions' must be specified")
-
-    if direction and directions:
-        raise ValueError("Cannot specify both 'direction' and 'directions'")
-
-    try:
-        # Initialize controller (which ensures workspace)
-        controller = AIAutoController(token)
-        
-        # Prepare request data for CreateStudy
-        request_data = {
-            "spec": {
-                "studyName": study_name,
-                "direction": direction or "",
-                "directions": directions or [],
-                "samplerJson": object_to_json(sampler),
-                "prunerJson": object_to_json(pruner)
-            }
-        }
-        
-        # Call CreateStudy RPC
-        response = controller.client.call_rpc("CreateStudy", request_data)
-        
-        # Return StudyWrapper
-        return StudyWrapper(
-            study_name=response.get("studyName", study_name),
-            storage=controller.storage,
-            controller=controller
-        )
-
-    except Exception as e:
-        raise RuntimeError(f"Failed to create study: {e}") from e

aiauto_client-0.1.4.dist-info/RECORD

@@ -1,11 +0,0 @@
-aiauto/__init__.py,sha256=TgD2ZvIHb7oKJb-HjUl3WfXXtuWLien0sybSy9onjL8,395
-aiauto/_config.py,sha256=DaRTIZlph9T3iuW-Cy4fkw8i3bXB--gMtW947SLZZNs,159
-aiauto/api.py,sha256=hzoVZMwKtH2EaAM1bY67grp6cenltCl8kdG8YHPegvk,1517
-aiauto/constants.py,sha256=UhDCLFoPE89XrHB3SEnZR3YUuzajgugMGX80KYx_qc0,939
-aiauto/core.py,sha256=dF-J8jxRP_FkQtxYYsrgza8OQHNzoPLN1cjYeUaqX1s,8076
-aiauto/http_client.py,sha256=t1gxeM5-d5bsVoFWgaNcTrt_WWUXuMuxge9gDlEqhoA,2086
-aiauto/serializer.py,sha256=_iPtEoqW8RTKOZ6UrC7CzOqoangpPYzeL7MQfIdmov8,1568
-aiauto_client-0.1.4.dist-info/METADATA,sha256=g86ft1ph8ebOUxQRafbgsYvxdlHRygOxvuS3vpSbfgQ,3001
-aiauto_client-0.1.4.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-aiauto_client-0.1.4.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
-aiauto_client-0.1.4.dist-info/RECORD,,