Fix type annotations in Input

replicate · Oct 16, 2024 · c2e1634 · c2e1634
1 parent 1e59833
commit c2e1634
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/python/cog/types.py b/python/cog/types.py
@@ -22,13 +22,13 @@
 
 def Input(
     default: Any = ...,
-    description: str = None,
-    ge: float = None,
-    le: float = None,
-    min_length: int = None,
-    max_length: int = None,
-    regex: str = None,
-    choices: List[Union[str, int]] = None,
+    description: Optional[str] = None,
+    ge: Optional[float] = None,
+    le: Optional[float] = None,
+    min_length: Optional[int] = None,
+    max_length: Optional[int] = None,
+    regex: Optional[str] = None,
+    choices: Optional[List[Union[str, int]]] = None,
 ) -> Any:
     """Input is similar to pydantic.Field, but doesn't require a default value to be the first argument."""
     return Field(