type

lucidrains · Sep 23, 2024 · 344cadd · 344cadd
1 parent f585a6a
commit 344cadd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alphafold3_pytorch/inputs.py b/alphafold3_pytorch/inputs.py
@@ -217,7 +217,7 @@
 
 @typecheck
 def maybe_cache(
-    fn,
+    fn: Callable,
     *,
     cache: dict,
     key: str | None,