Refactor torch training test to stop using metrics from checkpoint

daniil-lyakhov · Sep 15, 2023 · 8848d75 · 8848d75
1 parent 1e7a0c2
commit 8848d75
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/torch/test_compression_training.py b/tests/torch/test_compression_training.py
@@ -174,9 +174,7 @@ def finalize(self, dataset_dir, tmp_path_factory, weekly_models_path) -> "Compre
         return self
 
     def get_metric(self):
-        return self.sample_handler.get_metric_value_from_checkpoint(
-            self.checkpoint_save_dir, self.checkpoint_name, self.config_path
-        )
+        return self.expected_accuracy_
 
     def _get_weight_path(self, weekly_models_path):
         if self.weights_filename_ is None:
@@ -249,9 +247,7 @@ def subnet_expected_accuracy(self, subnet_expected_accuracy: float):
         return self
 
     def get_subnet_metric(self):
-        return self.sample_handler.get_metric_value_from_checkpoint(
-            self.checkpoint_save_dir, self.subnet_checkpoint_name
-        )
+        return self.subnet_expected_accuracy_
 
     def _get_weight_path(self, weekly_models_path):
         return os.path.join(