chore(run_mutagenesis): added print after model load #126

jyaacoub · Oct 24, 2024 · 81065e0 · 81065e0
1 parent d438058
commit 81065e0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/run_mutagenesis.py b/run_mutagenesis.py
@@ -110,6 +110,7 @@ def get_protein_features(pdb_file_path, cmap_thresh=8.0):
 pro, pdb_original = get_protein_features(PDB_FILE)
 original_seq = pdb_original.sequence
 
+print("Model Loaded")
 original_pkd = m(pro.to(DEVICE), lig.to(DEVICE))
 print("Original pkd:", original_pkd)