From ead737367c03d76f36e2293a368e579159f5ba97 Mon Sep 17 00:00:00 2001 From: Deven Navani Date: Mon, 5 Aug 2024 16:01:17 +0000 Subject: [PATCH] Try type: ignore --- modal/_utils/function_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modal/_utils/function_utils.py b/modal/_utils/function_utils.py index 13f1400ac..8c721bd2e 100644 --- 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"), }