aiauto-client 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

aiauto/serializer.py

@@ -6,14 +6,24 @@ from typing import Callable, Union, List, Optional
 def serialize(objective: Callable) -> str:
     try:
         return inspect.getsource(objective)
-    except Exception as e:
-        raise ValueError("objective는 모듈 최상위 def만 허용합니다(데코레이터/로컬/람다 불가)") from e
+    except (OSError, TypeError) as e:
+        raise ValueError(
+            "Serialize 실패 objective 함수는 파일로 저장되어야 합니다\n"
+            "REPL/Jupyter Notebook에서는 %%writefile magic을 사용해서 local 에 file 로 저장하세요\n\n"
+            "%%writefile objective.py\n"
+            "def objective(trial):\n"
+            "    # 함수 내용\n"
+            "    ...\n\n"
+            "그 다음:\n"
+            "from objective import objective\n"
+            "study.optimize(objective, ...)"
+        ) from e
 
 
 def build_requirements(file_path: Optional[str] = None, reqs: Optional[List[str]] = None) -> str:
     if file_path and reqs:
         raise ValueError("requirements_file과 requirements_list는 동시에 지정할 수 없습니다")
-    
+
     if file_path:
         with open(file_path, 'r') as f:
             return f.read()
@@ -26,20 +36,20 @@ def build_requirements(file_path: Optional[str] = None, reqs: Optional[List[str]
 def object_to_json(obj: Union[object, dict, None]) -> str:
     if obj is None:
         return ""
-    
+
     if isinstance(obj, dict):
         return json.dumps(obj)
-    
+
     cls = type(obj)
     module_name = cls.__module__
     class_name = cls.__name__
-    
+
     if not module_name.startswith('optuna.'):
         raise ValueError(f"optuna 코어 클래스만 지원합니다: {class_name}")
-    
+
     sig = inspect.signature(cls)
     kwargs = {}
-    
+
     for param_name, param in sig.parameters.items():
         if param_name == 'self':
             continue
@@ -47,9 +57,9 @@ def object_to_json(obj: Union[object, dict, None]) -> str:
             value = getattr(obj, param_name)
             if param.default != value:
                 kwargs[param_name] = value
-    
+
     return json.dumps({
         "module": module_name,
         "class": class_name,
         "kwargs": kwargs
-    })
+    })

{aiauto_client-0.1.6.dist-info → aiauto_client-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiauto-client
-Version: 0.1.6
+Version: 0.1.7
 Summary: AI Auto HPO (Hyperparameter Optimization) Client Library
 Author-email: AIAuto Team <ainode@zeroone.ai>
 Project-URL: Homepage, https://dashboard.aiauto.pangyo.ainode.ai
@@ -50,8 +50,8 @@ AIAuto는 Kubernetes 기반의 분산 HPO(Hyperparameter Optimization) 시스템
 - `parallelism` (int): 동시 실행 Pod 수 (기본값: 2)
 - `requirements_file` (str): requirements.txt 파일 경로 (requirements_list와 상호 배타적)
 - `requirements_list` (List[str]): 패키지 리스트 (requirements_file과 상호 배타적)
-- `resources_requests` (Dict[str, str]): K8s 리소스 요청 (기본값: {"cpu": "256m", "memory": "256Mi"})
-- `resources_limits` (Dict[str, str]): K8s 리소스 제한 (기본값: {"cpu": "256m", "memory": "256Mi"})
+- `resources_requests` (Dict[str, str]): 리소스 요청 (기본값: {"cpu": "256m", "memory": "256Mi"})
+- `resources_limits` (Dict[str, str]): 리소스 제한 (기본값: {"cpu": "256m", "memory": "256Mi"})
 - `runtime_image` (str): 커스텀 런타임 이미지 (None이면 자동 선택)
 - `use_gpu` (bool): GPU 사용 여부 (기본값: False)
 
@@ -95,6 +95,39 @@ study_wrapper.optimize(
 )
 ```
 
+## Jupyter Notebook 사용 시 주의사항
+
+Jupyter Notebook이나 Python REPL에서 정의한 함수는 Serialize 할 수 없습니다
+대신 `%%writefile` magic 울 사용하여 파일로 저장한 후 import 하세요.
+
+### Jupyter에서 objective 함수 작성 방법
+- objective 함수를 파일로 저장
+```python
+%%writefile my_objective.py
+import aiauto
+import optuna
+
+def objective(trial: optuna.trial.Trial):
+    """
+    이 함수는 외부 서버에서 실행됩니다.
+    모든 import는 함수 내부에 작성하세요.
+    """
+    import torch  # 함수 내부에서 import
+    
+    x = trial.suggest_float('x', -10, 10)
+    y = trial.suggest_float('y', -10, 10)
+    return (x - 2) ** 2 + (y - 3) ** 2
+```
+- 저장한 함수를 import해서 사용
+```python
+import aiauto
+from my_objective import objective
+
+ac = aiauto.AIAutoController('<token>')
+study = ac.create_study('test', 'minimize')
+study.optimize(objective, n_trials=10, parallelism=2)
+```
+
 ## 빠른 시작
 
 ### 1. 간단한 예제 (수학 함수 최적화)

{aiauto_client-0.1.6.dist-info → aiauto_client-0.1.7.dist-info}/RECORD

@@ -3,8 +3,8 @@ aiauto/_config.py,sha256=DaRTIZlph9T3iuW-Cy4fkw8i3bXB--gMtW947SLZZNs,159
 aiauto/constants.py,sha256=rBibGOQHHrdkwaai92-3I8-N0cu-B4CoCoQbG9-Cl8k,821
 aiauto/core.py,sha256=DQW9uvVNqP9J9s1IFx969upw10NQLHJMNsudHnauk6A,9840
 aiauto/http_client.py,sha256=t1gxeM5-d5bsVoFWgaNcTrt_WWUXuMuxge9gDlEqhoA,2086
-aiauto/serializer.py,sha256=_iPtEoqW8RTKOZ6UrC7CzOqoangpPYzeL7MQfIdmov8,1568
-aiauto_client-0.1.6.dist-info/METADATA,sha256=FOC7h1MkE6dlPa7urKnWX2sIUqe3Ac40TJNC1JgFzu4,17487
-aiauto_client-0.1.6.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
-aiauto_client-0.1.6.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
-aiauto_client-0.1.6.dist-info/RECORD,,
+aiauto/serializer.py,sha256=KqQeH0xp4LQuZE6r8kzXQsWY6QgC3hqn8MSuWTt4QmU,1938
+aiauto_client-0.1.7.dist-info/METADATA,sha256=xy8ZGkJYOwMUgA_8J4ag2mepwBzGZQwsyIOkMgQ8Cpw,18450
+aiauto_client-0.1.7.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
+aiauto_client-0.1.7.dist-info/top_level.txt,sha256=Sk2ctO9_Bf_tAPwq1x6Vfl6OuL29XzwMTO4F_KG6oJE,7
+aiauto_client-0.1.7.dist-info/RECORD,,