From 344caddc04a6c9782ac003f30e7a7cef6f4fb67f Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Mon, 23 Sep 2024 10:09:06 -0700
Subject: [PATCH] type

---
 alphafold3_pytorch/inputs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/alphafold3_pytorch/inputs.py b/alphafold3_pytorch/inputs.py
index 06158d3c..7817ac65 100644
--- 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,