Remove statscollector from GameError

gsvgit · Aug 30, 2024 · 3407414 · 3407414
1 parent 0ca57e6
commit 3407414
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/AIAgent/ml/game/errors_game.py b/AIAgent/ml/game/errors_game.py
@@ -1,4 +1,3 @@
-from ml.training.epochs_statistics import StatisticsCollector
 from connection.errors_connection import GameInterruptedError
 from common.utils import inheritors
 from common.game import GameMap2SVM
@@ -16,8 +15,6 @@ def __init__(
 
  super().__init__(game_map2svm, error_name)
 
- def handle_error(self, statistics_collector: StatisticsCollector):
+ def need_to_save_map(self):
  gie_inheritors = inheritors(GameInterruptedError)
- if self._error_name in map(lambda it: it.__name__, gie_inheritors):
- return
- statistics_collector.fail(self._map)
+ return self._error_name not in map(lambda it: it.__name__, gie_inheritors)
diff --git a/AIAgent/ml/training/validation.py b/AIAgent/ml/training/validation.py
@@ -80,7 +80,9 @@ def validate_coverage(
  colour=progress_bar_colour,
  ):
  if isinstance(result, GameError):
- result.handle_error(statistics_collector)
+ need_to_save_map: bool = result.need_to_save_map()
+ if not need_to_save_map:
+ statistics_collector.fail(result._map)
  else:
  all_results.append(result)