aiauto-client 0.1.16__py3-none-any.whl → 0.1.17__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- aiauto/core.py +3 -3
- {aiauto_client-0.1.16.dist-info → aiauto_client-0.1.17.dist-info}/METADATA +1 -1
- {aiauto_client-0.1.16.dist-info → aiauto_client-0.1.17.dist-info}/RECORD +5 -5
- {aiauto_client-0.1.16.dist-info → aiauto_client-0.1.17.dist-info}/WHEEL +0 -0
- {aiauto_client-0.1.16.dist-info → aiauto_client-0.1.17.dist-info}/top_level.txt +0 -0
aiauto/core.py
CHANGED
|
@@ -190,7 +190,7 @@ class StudyWrapper:
|
|
|
190
190
|
requirements_list: Optional[List[str]] = None,
|
|
191
191
|
resources_requests: Optional[Dict[str, str]] = None,
|
|
192
192
|
resources_limits: Optional[Dict[str, str]] = None,
|
|
193
|
-
runtime_image: Optional[str] =
|
|
193
|
+
runtime_image: Optional[str] = None,
|
|
194
194
|
use_gpu: bool = False
|
|
195
195
|
) -> None:
|
|
196
196
|
# 리소스 기본값 설정
|
|
@@ -199,13 +199,13 @@ class StudyWrapper:
|
|
|
199
199
|
resources_requests = {"cpu": "2", "memory": "4Gi"}
|
|
200
200
|
else:
|
|
201
201
|
resources_requests = {"cpu": "1", "memory": "1Gi"}
|
|
202
|
-
|
|
202
|
+
|
|
203
203
|
if resources_limits is None:
|
|
204
204
|
if use_gpu:
|
|
205
205
|
resources_limits = {"cpu": "2", "memory": "4Gi"}
|
|
206
206
|
else:
|
|
207
207
|
resources_limits = {"cpu": "1", "memory": "1Gi"}
|
|
208
|
-
|
|
208
|
+
|
|
209
209
|
if runtime_image is None or runtime_image == "":
|
|
210
210
|
if use_gpu:
|
|
211
211
|
runtime_image = "pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime"
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: aiauto-client
|
|
3
|
-
Version: 0.1.
|
|
3
|
+
Version: 0.1.17
|
|
4
4
|
Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
|
|
5
5
|
Author-email: AIAuto Team <ainode@zeroone.ai>
|
|
6
6
|
Project-URL: Homepage, https://dashboard.common.aiauto.pangyo.ainode.ai
|
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
aiauto/__init__.py,sha256=sF7sJaXg7-MqolSYLxsaXAir1dBzARhXLrHo7zLsupg,345
|
|
2
2
|
aiauto/_config.py,sha256=hTFh2bH9m-HuX6QCpNtBC0j6rEB0S97hhPKjbEjv4Tg,89
|
|
3
3
|
aiauto/constants.py,sha256=rBibGOQHHrdkwaai92-3I8-N0cu-B4CoCoQbG9-Cl8k,821
|
|
4
|
-
aiauto/core.py,sha256=
|
|
4
|
+
aiauto/core.py,sha256=z46oRcuwRR4xfqKbbokqB13vo29ayLxjZeGi8zAKNuA,10306
|
|
5
5
|
aiauto/http_client.py,sha256=v_nPdb-2tIeH1XrOYqzMGvFfXLKEDbQoSaQYPsB0Hik,2587
|
|
6
6
|
aiauto/serializer.py,sha256=BJmeq6uCD9D2_6bXu_sMBQLSsXCUMIMM10iX923DTXE,2749
|
|
7
|
-
aiauto_client-0.1.
|
|
8
|
-
aiauto_client-0.1.
|
|
9
|
-
aiauto_client-0.1.
|
|
10
|
-
aiauto_client-0.1.
|
|
7
|
+
aiauto_client-0.1.17.dist-info/METADATA,sha256=rpaVYPlOncrSHbTpSykm3cNWoQFjsvOrmThkkCz-pFU,25625
|
|
8
|
+
aiauto_client-0.1.17.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
|
|
9
|
+
aiauto_client-0.1.17.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
|
|
10
|
+
aiauto_client-0.1.17.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|