20220511 metric change 2

openvinotoolkit · May 31, 2022 · 184df04 · 184df04
1 parent 3bac4de
commit 184df04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/external/anomaly/ote_anomalib/callbacks/progress.py b/external/anomaly/ote_anomalib/callbacks/progress.py
@@ -117,7 +117,7 @@ def on_validation_epoch_end(self, trainer, pl_module):
         super().on_validation_epoch_end(trainer, pl_module)
         score = None
         metric = getattr(self.update_progress_callback, 'metric', None)
-        if metric in list(trainer.logged_metrics.keys()):
+        if metric in trainer.logged_metrics:
             score = float(trainer.logged_metrics[metric])
         progress = int(self._get_progress('predict'))
         self.update_progress_callback(progress=progress, score=score)