aiauto-client 0.1.4__tar.gz → 0.1.5__tar.gz

{aiauto_client-0.1.4 → aiauto_client-0.1.5}/PKG-INFO

@@ -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.4 → aiauto_client-0.1.5}/pyproject.toml

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

{aiauto_client-0.1.4 → aiauto_client-0.1.5}/src/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_client-0.1.4 → aiauto_client-0.1.5}/src/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 → aiauto_client-0.1.5}/src/aiauto_client.egg-info/PKG-INFO

@@ -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.4 → aiauto_client-0.1.5}/src/aiauto_client.egg-info/SOURCES.txt

@@ -2,7 +2,6 @@ README.md
 pyproject.toml
 src/aiauto/__init__.py
 src/aiauto/_config.py
-src/aiauto/api.py
 src/aiauto/constants.py
 src/aiauto/core.py
 src/aiauto/http_client.py

aiauto_client-0.1.4/src/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