Fix bad import

womboai · Dec 19, 2024 · d425e6f · d425e6f
1 parent d507abc
commit d425e6f
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/validator/weight_setting/contest_state.py b/validator/weight_setting/contest_state.py
@@ -6,9 +6,8 @@
 
 from base.checkpoint import Key, current_time, Submissions, Benchmarks
 from base.contest import Metrics, BenchmarkState
-from weight_setting.winner_selection import get_contestant_scores, get_contestant_ranks, calculate_rank_weights
-
-from inputs_api import get_inputs_state
+from base.inputs_api import get_inputs_state
+from weight_setting.winner_selection import get_contestant_scores
 from winner_selection import calculate_score_weights
 
 logger = get_logger(__name__)
@@ -68,9 +67,6 @@ def sleep_to_next_contest(self, stop_flag: Event):
         stop_flag.wait(next_contest_time.total_seconds())
 
     def get_scores(self, benchmarks: Benchmarks) -> dict[Key, float]:
-        if not self.baseline:
-            return {}
-
         return get_contestant_scores(
             submissions=self.submissions,
             benchmarks=benchmarks,