minor cleanup

dscript/__main__.py

@@ -5,7 +5,15 @@ D-SCRIPT: Structure Aware PPI Prediction
 import argparse
 import sys
 
-from .commands import predict_serial
+from .commands import (
+    embed,
+    evaluate,
+    extract_3di,
+    predict_bipartite,
+    predict_block,
+    predict_serial,
+    train,
+)
 from .commands.embed import EmbeddingArguments
 from .commands.evaluate import EvaluateArguments
 from .commands.extract_3di import Extract3DiArguments
@@ -56,15 +64,6 @@ def main():
     subparsers = parser.add_subparsers(title="D-SCRIPT Commands", dest="cmd")
     subparsers.required = True
 
-    from .commands import (
-        embed,
-        evaluate,
-        extract_3di,
-        predict_bipartite,
-        predict_block,
-        train,
-    )
-
     modules = {
         "train": train,
         "embed": embed,

dscript/commands/evaluate.py

@@ -17,7 +17,6 @@ import numpy as np
 import pandas as pd
 import torch
 from Bio import SeqIO
-from loguru import logger
 from sklearn.metrics import (
     average_precision_score,
     precision_recall_curve,
@@ -280,9 +279,6 @@ def main(args):
     phats = np.array(phats)
     labels = np.array(labels)
 
-    logger.debug(phats)
-    logger.debug(labels)
-
     with open(outPath + "_metrics.txt", "w+") as f:
         aupr = average_precision_score(labels, phats)
         auroc = roc_auc_score(labels, phats)