fix warnings

DeepRank · Sep 28, 2023 · 8dce49b · 8dce49b
1 parent 768405a
commit 8dce49b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/deeprank_gnn/predict.py b/src/deeprank_gnn/predict.py
@@ -39,10 +39,8 @@
 # Constants
 # TODO: Make these configurable
 ESM_MODEL = "esm2_t33_650M_UR50D"
-GNN_ESM_MODEL = (
-    Path(__file__).parent
-    / "paper_pretrained_models/scoring_of_docking_models/gnn_esm/treg_yfnat_b64_e20_lr0.001_foldall_esm.pth.tar"
-)
+GNN_ESM_MODEL = "paper_pretrained_models/scoring_of_docking_models/gnn_esm/treg_yfnat_b64_e20_lr0.001_foldall_esm.pth.tar"
+
 TOKS_PER_BATCH = 4096
 REPR_LAYERS = [0, 32, 33]
 TRUNCATION_SEQ_LENGTH = 2500