modal 0.73.157__py3-none-any.whl → 0.73.158__py3-none-any.whl

modal/_serialization.py

@@ -501,20 +501,20 @@ def _signature_parameter_to_spec(
         has_default=has_default,
     )
     if include_legacy_parameter_fields:
+        # Specific to *class parameters*:
         # add the .{type}_default and `.type` values as required by legacy clients
         # looking at class parameter specs
-        if full_proto_type.base_type == api_pb2.PARAM_TYPE_INT:
-            if has_default:
+        field_spec.type = field_spec.full_type.base_type
+
+        if has_default:
+            if full_proto_type.base_type == api_pb2.PARAM_TYPE_INT:
                 field_spec.int_default = python_signature_parameter.default
-            field_spec.type = api_pb2.PARAM_TYPE_INT
-        elif full_proto_type.base_type == api_pb2.PARAM_TYPE_STRING:
-            if has_default:
+            elif full_proto_type.base_type == api_pb2.PARAM_TYPE_STRING:
                 field_spec.string_default = python_signature_parameter.default
-            field_spec.type = api_pb2.PARAM_TYPE_STRING
-        elif full_proto_type.base_type == api_pb2.PARAM_TYPE_BYTES:
-            if has_default:
+            elif full_proto_type.base_type == api_pb2.PARAM_TYPE_BYTES:
                 field_spec.bytes_default = python_signature_parameter.default
-            field_spec.type = api_pb2.PARAM_TYPE_BYTES
+            elif full_proto_type.base_type == api_pb2.PARAM_TYPE_BOOL:
+                field_spec.bool_default = python_signature_parameter.default
 
     return field_spec
 

modal/_type_manager.py

@@ -143,6 +143,23 @@ class BytesParameter:
             raise TypeError(f"Expected bytes, got {type(python_value).__name__}")
 
 
+@parameter_serde_registry.register_encoder(bool)
+@parameter_serde_registry.register_decoder(api_pb2.PARAM_TYPE_BOOL)
+class BoolParameter:
+    @staticmethod
+    def encode(value: Any) -> api_pb2.ClassParameterValue:
+        return api_pb2.ClassParameterValue(type=api_pb2.PARAM_TYPE_BOOL, bool_value=value)
+
+    @staticmethod
+    def decode(proto_value: api_pb2.ClassParameterValue) -> bool:
+        return proto_value.bool_value
+
+    @staticmethod
+    def validate(python_value: Any):
+        if not isinstance(python_value, bool):
+            raise TypeError(f"Expected bool, got {type(python_value).__name__}")
+
+
 SCHEMA_FACTORY_TYPE = typing.Callable[[type], api_pb2.GenericPayloadType]
 
 
@@ -205,6 +222,13 @@ def str_schema(full_python_type: type) -> api_pb2.GenericPayloadType:
     )
 
 
+@schema_registry.add(bool)
+def bool_schema(full_python_type: type) -> api_pb2.GenericPayloadType:
+    return api_pb2.GenericPayloadType(
+        base_type=api_pb2.PARAM_TYPE_BOOL,
+    )
+
+
 @schema_registry.add(type(None))
 def none_type_schema(declared_python_type: type) -> api_pb2.GenericPayloadType:
     return api_pb2.GenericPayloadType(base_type=api_pb2.PARAM_TYPE_NONE)

modal/client.pyi

@@ -31,7 +31,7 @@ class _Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "0.73.157",
+        version: str = "0.73.158",
     ): ...
     def is_closed(self) -> bool: ...
     @property
@@ -93,7 +93,7 @@ class Client:
         server_url: str,
         client_type: int,
         credentials: typing.Optional[tuple[str, str]],
-        version: str = "0.73.157",
+        version: str = "0.73.158",
     ): ...
     def is_closed(self) -> bool: ...
     @property

modal/functions.pyi

@@ -200,11 +200,11 @@ class Function(
 
     _call_generator_nowait: ___call_generator_nowait_spec[typing_extensions.Self]
 
-    class __remote_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER, SUPERSELF]):
+    class __remote_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER, SUPERSELF]):
         def __call__(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> ReturnType_INNER: ...
         async def aio(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> ReturnType_INNER: ...
 
-    remote: __remote_spec[modal._functions.P, modal._functions.ReturnType, typing_extensions.Self]
+    remote: __remote_spec[modal._functions.ReturnType, modal._functions.P, typing_extensions.Self]
 
     class __remote_gen_spec(typing_extensions.Protocol[SUPERSELF]):
         def __call__(self, *args, **kwargs) -> typing.Generator[typing.Any, None, None]: ...
@@ -219,19 +219,19 @@ class Function(
         self, *args: modal._functions.P.args, **kwargs: modal._functions.P.kwargs
     ) -> modal._functions.OriginalReturnType: ...
 
-    class ___experimental_spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER, SUPERSELF]):
+    class ___experimental_spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER, SUPERSELF]):
         def __call__(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
         async def aio(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
 
     _experimental_spawn: ___experimental_spawn_spec[
-        modal._functions.P, modal._functions.ReturnType, typing_extensions.Self
+        modal._functions.ReturnType, modal._functions.P, typing_extensions.Self
     ]
 
-    class __spawn_spec(typing_extensions.Protocol[P_INNER, ReturnType_INNER, SUPERSELF]):
+    class __spawn_spec(typing_extensions.Protocol[ReturnType_INNER, P_INNER, SUPERSELF]):
         def __call__(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
         async def aio(self, *args: P_INNER.args, **kwargs: P_INNER.kwargs) -> FunctionCall[ReturnType_INNER]: ...
 
-    spawn: __spawn_spec[modal._functions.P, modal._functions.ReturnType, typing_extensions.Self]
+    spawn: __spawn_spec[modal._functions.ReturnType, modal._functions.P, typing_extensions.Self]
 
     def get_raw_f(self) -> collections.abc.Callable[..., typing.Any]: ...
 

{modal-0.73.157.dist-info → modal-0.73.158.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: modal
-Version: 0.73.157
+Version: 0.73.158
 Summary: Python client library for Modal
 Author-email: Modal Labs <support@modal.com>
 License: Apache-2.0

{modal-0.73.157.dist-info → modal-0.73.158.dist-info}/RECORD

@@ -13,17 +13,17 @@ modal/_proxy_tunnel.py,sha256=gnKyCfmVB7x2d1A6c-JDysNIP3kEFxmXzhcXhPrzPn0,1906
 modal/_pty.py,sha256=JZfPDDpzqICZqtyPI_oMJf_9w-p_lLNuzHhwhodUXio,1329
 modal/_resolver.py,sha256=RtoXoYzSllPlFu0D1vel_FWiEmDO7RyToiC2bxeN8ZY,6917
 modal/_resources.py,sha256=5qmcirXUI8dSH926nwkUaeX9H25mqYu9mXD_KuT79-o,1733
-modal/_serialization.py,sha256=kkLmQ6jnPlbLT3FfuyigocHKG7H1HDJ0i24IxYFbgs8,22927
+modal/_serialization.py,sha256=wAgaALThfr-DBV9LMhM4qY_PCH7SRhA9xgoHL2bapBk,22963
 modal/_traceback.py,sha256=IZQzB3fVlUfMHOSyKUgw0H6qv4yHnpyq-XVCNZKfUdA,5023
 modal/_tunnel.py,sha256=zTBxBiuH1O22tS1OliAJdIsSmaZS8PlnifS_6S5z-mk,6320
 modal/_tunnel.pyi,sha256=JmmDYAy9F1FpgJ_hWx0xkom2nTOFQjn4mTPYlU3PFo4,1245
-modal/_type_manager.py,sha256=SvDvX9JK1jLp8TA_psLd-hNYPALlG9KtSOmlbHqdtQE,8284
+modal/_type_manager.py,sha256=DWjgmjYJuOagw2erin506UUbG2H5UzZCFEekS-7hmfA,9087
 modal/_watcher.py,sha256=K6LYnlmSGQB4tWWI9JADv-tvSvQ1j522FwT71B51CX8,3584
 modal/app.py,sha256=w00bV9cjABAsS2ExE7zb1jY6Q_snXYmdKa3xRFg8iXA,47428
 modal/app.pyi,sha256=pUEqciyGZ446sc_QoG8XcQ_oc6oU-U4dqjkxjhgOX98,26968
 modal/call_graph.py,sha256=1g2DGcMIJvRy-xKicuf63IVE98gJSnQsr8R_NVMptNc,2581
 modal/client.py,sha256=U-YKSw0n7J1ZLREt9cbEJCtmHe5YoPKFxl0xlkan2yc,15565
-modal/client.pyi,sha256=2r_RawoavNS1KIGw_fAUQHoOIYkZni6yaea8_qedhHE,7661
+modal/client.pyi,sha256=2xDnuSzcyAojtKfIiW4h13jQXTDGW_VqMrToQ_VwKVM,7661
 modal/cloud_bucket_mount.py,sha256=YOe9nnvSr4ZbeCn587d7_VhE9IioZYRvF9VYQTQux08,5914
 modal/cloud_bucket_mount.pyi,sha256=30T3K1a89l6wzmEJ_J9iWv9SknoGqaZDx59Xs-ZQcmk,1607
 modal/cls.py,sha256=jlQE7qEwGG49FCSqqzYkcedi619G3K4PNnJCtPO8y_w,32954
@@ -40,7 +40,7 @@ modal/file_io.py,sha256=lcMs_E9Xfm0YX1t9U2wNIBPnqHRxmImqjLW1GHqVmyg,20945
 modal/file_io.pyi,sha256=NTRft1tbPSWf9TlWVeZmTlgB5AZ_Zhu2srWIrWr7brk,9445
 modal/file_pattern_matcher.py,sha256=trosX-Bp7dOubudN1bLLhRAoidWy1TcoaR4Pv8CedWw,6497
 modal/functions.py,sha256=kcNHvqeGBxPI7Cgd57NIBBghkfbeFJzXO44WW0jSmao,325
-modal/functions.pyi,sha256=m1PL2pwO-lnGV0uZDVCmzZ_v7Mu8ISRtxmxS15aEIAQ,14785
+modal/functions.pyi,sha256=Bg_zlYSHMYuEXQYLcFU-j1Kwq5zHQ9jggMA_5ZO_du8,14785
 modal/gpu.py,sha256=Kbhs_u49FaC2Zi0TjCdrpstpRtT5eZgecynmQi5IZVE,6752
 modal/image.py,sha256=HtkKomhX4inozqSRi7lf5Vt9IEqCnVHn5bEo59hD64A,92835
 modal/image.pyi,sha256=iWclz2rxaP-LSsYMgU0X3ZcN5mEFvpyKzIPKJbohmsg,25591
@@ -153,10 +153,10 @@ modal_docs/mdmd/__init__.py,sha256=svYKtV8HDwDCN86zbdWqyq5T8sMdGDj0PVlzc2tIxDM,2
 modal_docs/mdmd/mdmd.py,sha256=Irx49MCCTlBOP4FBdLR--JrpA3-WhsVeriq0LGgsRic,6232
 modal_docs/mdmd/signatures.py,sha256=XJaZrK7Mdepk5fdX51A8uENiLFNil85Ud0d4MH8H5f0,3218
 modal_proto/__init__.py,sha256=MIEP8jhXUeGq_eCjYFcqN5b1bxBM4fdk0VESpjWR0fc,28
-modal_proto/api.proto,sha256=id2Ep7N4h09b-TWSk8sBEJ7jDxnqTJR1EDzDQw_idtY,91357
+modal_proto/api.proto,sha256=VwW2t0kpo9FMrYq1jHQpL4pvLbes2oCUxw7EXKAplpA,91432
 modal_proto/api_grpc.py,sha256=9Rs0JyHcz_DSjVKhdtMbDuNt6qDkrE2718KsyA3QL4c,110702
-modal_proto/api_pb2.py,sha256=2PaWyzRrd5TIHN2imLlsiAaBpzqqHM1UXaj_t0h9k5A,321857
-modal_proto/api_pb2.pyi,sha256=S2p4cY5k8Uw4nxncuTzVLDftKldmnhCUsQtgA4MICKo,438482
+modal_proto/api_pb2.py,sha256=JXOC0c4UtDJkNfpwTJTB-rkVoeW1pMHTxr6iAb9ckQI,322009
+modal_proto/api_pb2.pyi,sha256=iJH5M3mh9TGO2nZV2Te0vfLawharrzxBsIndsoF9pWs,438965
 modal_proto/api_pb2_grpc.py,sha256=olXvs6OQvy7pqvHP9bkSWC_DdIv0iO38xRlmkLo-ai8,239213
 modal_proto/api_pb2_grpc.pyi,sha256=ybhcN2nwFBIPd4Z4kkMOv-M8Ejidz93Bl4zScLpYcK0,55706
 modal_proto/modal_api_grpc.py,sha256=43ujbC_a8YAjuhtEvS-O-5lNpkG5d0K0ZIlryJ4weT4,14766
@@ -170,10 +170,10 @@ modal_proto/options_pb2_grpc.pyi,sha256=CImmhxHsYnF09iENPoe8S4J-n93jtgUYD2JPAc0y
 modal_proto/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modal_version/__init__.py,sha256=wiJQ53c-OMs0Xf1UeXOxQ7FwlV1VzIjnX6o-pRYZ_Pk,470
 modal_version/__main__.py,sha256=2FO0yYQQwDTh6udt1h-cBnGd1c4ZyHnHSI4BksxzVac,105
-modal_version/_version_generated.py,sha256=8z1dZXLk13pbLXOUQfQfB2qHzbM_QT3SY6V9WVjVZAk,150
-modal-0.73.157.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
-modal-0.73.157.dist-info/METADATA,sha256=GUT8XtlnOYHmpHyo6u1Q5XlsoEOPQGyuJB-UVeLDm6w,2453
-modal-0.73.157.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-modal-0.73.157.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
-modal-0.73.157.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
-modal-0.73.157.dist-info/RECORD,,
+modal_version/_version_generated.py,sha256=W-Cjy5h17CL7l8Q5DFjspeFnQc1xP0KQ83zpkDXqVD4,150
+modal-0.73.158.dist-info/LICENSE,sha256=psuoW8kuDP96RQsdhzwOqi6fyWv0ct8CR6Jr7He_P_k,10173
+modal-0.73.158.dist-info/METADATA,sha256=V8G3VKDYed7Y6hnUEmwQNtFOrBO-0ORRHED2hfH9yxk,2453
+modal-0.73.158.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+modal-0.73.158.dist-info/entry_points.txt,sha256=An-wYgeEUnm6xzrAP9_NTSTSciYvvEWsMZILtYrvpAI,46
+modal-0.73.158.dist-info/top_level.txt,sha256=4BWzoKYREKUZ5iyPzZpjqx4G8uB5TWxXPDwibLcVa7k,43
+modal-0.73.158.dist-info/RECORD,,

modal_proto/api.proto

@@ -176,6 +176,7 @@ enum ParameterType {
   PARAM_TYPE_LIST = 6;
   PARAM_TYPE_DICT = 7;
   PARAM_TYPE_NONE = 8;
+  PARAM_TYPE_BOOL = 9;
 }
 
 enum ProgressType {
@@ -710,6 +711,7 @@ message ClassParameterSpec { // TODO: rename into NamedPayloadType or similar
     int64 int_default = 5;
     bytes pickle_default = 6;
     bytes bytes_default = 7;
+    bool bool_default = 9;
   }
   GenericPayloadType full_type = 8; // supersedes `type`
 }
@@ -726,6 +728,7 @@ message ClassParameterValue {  // TODO: rename into NamedPayloadValue
     int64 int_value = 4;
     bytes pickle_value = 5;
     bytes bytes_value = 6;
+    bool bool_value = 7;
   }
 }