Revert type fix

modal-labs · Aug 5, 2024 · 7d4158c · 7d4158c
1 parent cbc3351
commit 7d4158c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modal/_utils/function_utils.py b/modal/_utils/function_utils.py
@@ -28,7 +28,7 @@ class FunctionInfoType(Enum):
     NOTEBOOK = "notebook"
 
 
-CLASS_PARAM_TYPE_MAP: Dict[Type, Tuple[api_pb2.ParameterType, str]] = {
+CLASS_PARAM_TYPE_MAP: Dict[Type, Tuple[api_pb2.ParameterType.ValueType, str]] = {
     str: (api_pb2.PARAM_TYPE_STRING, "string_default"),
     int: (api_pb2.PARAM_TYPE_INT, "int_default"),
 }