Make export function preserve types

chanzuckerberg · Dec 14, 2023 · 5731f85 · 5731f85
1 parent 3f684ea
commit 5731f85
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/WDL/_util.py b/WDL/_util.py
@@ -32,10 +32,12 @@
 from types import FrameType
 from pythonjsonlogger import jsonlogger
 
+T = TypeVar("T")
+
 __all__: List[str] = []
 
 
-def export(obj) -> str:  # pyre-ignore
+def export(obj: T) -> T:  # pyre-ignore
     __all__.append(obj.__name__)
     return obj
 
@@ -67,7 +69,6 @@ def strip_leading_whitespace(txt: str) -> Tuple[int, str]:
     return (to_strip, "\n".join(lines))
 
 
-T = TypeVar("T")
 
 
 @export