Update alphafold3.py (#118)

amorehead · web-flow · commit fe0d7f0b1e20 · 2024-08-01T18:33:17.000-07:00
diff --git a/alphafold3_pytorch/alphafold3.py b/alphafold3_pytorch/alphafold3.py
@@ -25,6 +25,7 @@
     Float,
     Int,
     Bool,
+    Shaped,
     typecheck
 )
 
