Try type: ignore

modal-labs · Aug 5, 2024 · ead7373 · ead7373
1 parent 64719c5
commit ead7373
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.V, str]] = {
+CLASS_PARAM_TYPE_MAP: Dict[Type, Tuple[api_pb2.ParameterType.ValueType, str]] = {  # type: ignore
     str: (api_pb2.PARAM_TYPE_STRING, "string_default"),
     int: (api_pb2.PARAM_TYPE_INT, "int_default"),
 }