move into one file, cleanup

Author: lucidrains

alphafold3_pytorch/__init__.py

@@ -32,6 +32,7 @@
     Alphafold3,
     Alphafold3WithHubMixin,
     ConfidenceHeadLogits,
+    ComputeRankingScore
 )
 
 from alphafold3_pytorch.inputs import (
@@ -67,10 +68,6 @@
     create_trainer_from_conductor_yaml
 )
 
-from alphafold3_pytorch.confidence import (
-    ComputeRankingScore,
-)
-
 __all__ = [
     Attention,
     Attend,