diff --git a/dscript/__main__.py b/dscript/__main__.py index 731d7a1..812beea 100644 --- a/dscript/__main__.py +++ b/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, diff --git a/dscript/commands/evaluate.py b/dscript/commands/evaluate.py index 9a0df06..37f5d03 100644 --- a/dscript/commands/evaluate.py +++ b/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)