feat: Prints model names while running

2022-08-13 11:10:00 -05:00 · 2022-08-13 11:10:00 -05:00 · 7ea1d715f6
commit 7ea1d715f6
parent ed102799d1
1 changed files with 4 additions and 2 deletions
--- a/deepdog/real_spectrum_run.py
+++ b/deepdog/real_spectrum_run.py
@ -140,8 +140,10 @@ class RealSpectrumRun:

 		results = []
 		_logger.debug("Going to iterate over models now")
-		for model_count, model in enumerate(self.models):
-			_logger.debug(f"Doing model #{model_count}")
+		for model_count, (model, model_name) in enumerate(
+			zip(self.models, self.model_names)
+		):
+			_logger.debug(f"Doing model #{model_count}: {model_name}")
 			core_count = multiprocessing.cpu_count() - 1 or 1
 			with multiprocessing.Pool(core_count) as pool:
 				cycle_count = 0