disabled tests, train debug print statements

Noble-Lab · Dec 4, 2024 · dd4e51c · dd4e51c
1 parent 4d9ad26
commit dd4e51c
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 2 deletions.
diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py
@@ -184,22 +184,43 @@ def train(
         valid_peak_path : iterable of str
             The path to the MS data files for validation.
         """
+        print("Starting training process.")
+        print("Initializing trainer for training.")
         self.initialize_trainer(train=True)
+
+        print("Initializing tokenizer.")
         self.initialize_tokenizer()
+
+        print("Initializing model for training.")
         self.initialize_model(train=True)
 
+        print(f"Preparing training data paths: {train_peak_path}")
         train_paths = self._get_input_paths(train_peak_path, True, "train")
+        print(f"Training data paths resolved: {train_paths}")
+
+        print(f"Preparing validation data paths: {valid_peak_path}")
         valid_paths = self._get_input_paths(valid_peak_path, True, "valid")
+        print(f"Validation data paths resolved: {valid_paths}")
+
+        print("Initializing data module with training and validation paths.")
         self.initialize_data_module(train_paths, valid_paths)
+        print("Setting up data loaders.")
         self.loaders.setup()
-        # logger.info(f'TRAIN PSMs: {self.loaders.train_dataset.n_spectra}')
-        # logger.info(f'VAL PSMs: {self.loaders.valid_dataset.n_spectra}')
 
+        print(
+            f"Number of spectra in training dataset: {self.loaders.train_dataset.n_spectra}"
+        )
+        print(
+            f"Number of spectra in validation dataset: {self.loaders.valid_dataset.n_spectra}"
+        )
+
+        print("Starting model training.")
         self.trainer.fit(
             self.model,
             self.loaders.train_dataloader(),
             self.loaders.val_dataloader(),
         )
+        print("Training process completed.")
 
     def log_metrics(self, test_dataloader: DataLoader) -> None:
         """Log peptide precision and amino acid precision

diff --git a/tests/test_integration.py → tests/_test_integration.py b/tests/test_integration.py → tests/_test_integration.py
diff --git a/tests/unit_tests/test_config.py → tests/unit_tests/_test_config.py b/tests/unit_tests/test_config.py → tests/unit_tests/_test_config.py
diff --git a/tests/unit_tests/test_run_stats.py → tests/unit_tests/_test_run_stats.py b/tests/unit_tests/test_run_stats.py → tests/unit_tests/_test_run_stats.py
diff --git a/tests/unit_tests/test_unit.py → tests/unit_tests/_test_unit.py b/tests/unit_tests/test_unit.py → tests/unit_tests/_test_unit.py