From 5d089a1cca5dd3035310b043fd5cc46bcbab0c63 Mon Sep 17 00:00:00 2001
From: justin-a-sanders <60298590+justin-a-sanders@users.noreply.github.com>
Date: Wed, 13 Mar 2024 00:28:17 -0700
Subject: [PATCH] remove time profiling

---
 casanovo/denovo/model_runner.py | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py
index 72cd78b7..e23a1b42 100644
--- a/casanovo/denovo/model_runner.py
+++ b/casanovo/denovo/model_runner.py
@@ -20,12 +20,8 @@
 from ..denovo.model import Spec2Pep
 from ..denovo.model import DBSpec2Pep
 
-from pytorch_lightning.profiler import SimpleProfiler, AdvancedProfiler
-
-
 logger = logging.getLogger("casanovo")
 
-
 def predict(
     peak_path: str,
     model_filename: str,
@@ -393,10 +389,7 @@ def db_search(
 
     # Create the Trainer object.
     abs_experiment_dirpath = "/net/noble/vol2/home/vananth3/2023_vananth_denovo-dbsearch/results/2023-08-21_speedup"
-    profiler = AdvancedProfiler(
-        dirpath=abs_experiment_dirpath,
-        filename="placeholder",
-    )
+
     trainer = pl.Trainer(
         accelerator="auto",
         auto_select_gpus=True,
@@ -404,8 +397,7 @@ def db_search(
         logger=config["logger"],
         max_epochs=config["max_epochs"],
         num_sanity_val_steps=config["num_sanity_val_steps"],
-        strategy=_get_strategy(),
-        profiler=profiler,
+        strategy=_get_strategy()
     )
     # Run the model
     trainer.predict(model, loaders.db_dataloader())