aiauto-client 0.1.16__tar.gz → 0.1.17__tar.gz

{aiauto_client-0.1.16 → aiauto_client-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.16
+Version: 0.1.17
 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.16 → aiauto_client-0.1.17}/examples/example_torch_single_objective.py

@@ -155,10 +155,10 @@ def objective_single(trial):
         root="/tmp/cifar10_data",  # Pod의 임시 디렉토리 사용
         train=True,
         download=True,
-        transform=[
+        transform=transforms.Compose([
             transforms.ToTensor(),
             transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
-        ],
+        ]),
     )
     # 전체 데이터를 train, valid로 분할
     n_total = len(dataset)
@@ -355,19 +355,19 @@ def objective_detailed(trial):
         root="/tmp/cifar10_data",
         train=True,
         download=True,
-        transform=[
+        transform=transforms.Compose([
             transforms.ToTensor(),
             transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
-        ],
+        ]),
     )
     test_dataset = datasets.CIFAR10(
         root="/tmp/cifar10_data",
         train=False,
         download=True,
-        transform=[
+        transform=transforms.Compose([
             transforms.ToTensor(),
             transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
-        ],
+        ]),
     )
 
     tc.log(f'full dataset: train {len(train_dataset)}, test {len(test_dataset)}')

{aiauto_client-0.1.16 → aiauto_client-0.1.17}/pyproject.toml

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

{aiauto_client-0.1.16 → aiauto_client-0.1.17}/src/aiauto/core.py

@@ -190,7 +190,7 @@ class StudyWrapper:
         requirements_list: Optional[List[str]] = None,
         resources_requests: Optional[Dict[str, str]] = None,
         resources_limits: Optional[Dict[str, str]] = None,
-        runtime_image: Optional[str] = 'ghcr.io/astral-sh/uv:python3.8-bookworm-slim',
+        runtime_image: Optional[str] = None,
         use_gpu: bool = False
     ) -> None:
         # 리소스 기본값 설정
@@ -199,13 +199,13 @@ class StudyWrapper:
                 resources_requests = {"cpu": "2", "memory": "4Gi"}
             else:
                 resources_requests = {"cpu": "1", "memory": "1Gi"}
-        
+
         if resources_limits is None:
             if use_gpu:
                 resources_limits = {"cpu": "2", "memory": "4Gi"}
             else:
                 resources_limits = {"cpu": "1", "memory": "1Gi"}
-        
+
         if runtime_image is None or runtime_image == "":
             if use_gpu:
                 runtime_image = "pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime"

{aiauto_client-0.1.16 → aiauto_client-0.1.17}/src/aiauto_client.egg-info/PKG-INFO

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