fix tests

lucidrains · lucidrains · commit 1e3c221036e5 · 2024-05-25T08:01:53.000-07:00
diff --git a/alphafold3_pytorch/alphafold3.py b/alphafold3_pytorch/alphafold3.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 """
 global ein notation:
 
@@ -33,8 +35,6 @@
 9: is_ligand
 """
 
-from __future__ import annotations
-
 from math import pi, sqrt
 from functools import partial, wraps
 from collections import namedtuple
