Skip to content

Commit 16bfc7e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent e35d554 commit 16bfc7e

1 file changed

Lines changed: 10 additions & 12 deletions

File tree

tests/unit/amlb/test_results_race_condition.py

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,11 @@
11
"""Test for race condition fix in local results file writing (issue #691)."""
22

3-
import os
43
import tempfile
54
from concurrent.futures import ThreadPoolExecutor
65
from pathlib import Path
76

8-
import pytest
97

10-
from amlb.results import Scoreboard, TaskResult
8+
from amlb.results import Scoreboard
119

1210

1311
def test_parallel_save_no_race_condition():
@@ -63,23 +61,23 @@ def save_board(board):
6361
result_df = result_board.as_data_frame()
6462

6563
# Check that we have all the expected rows
66-
assert (
67-
len(result_df) == num_parallel_saves
68-
), f"Expected {num_parallel_saves} rows, but got {len(result_df)}"
64+
assert len(result_df) == num_parallel_saves, (
65+
f"Expected {num_parallel_saves} rows, but got {len(result_df)}"
66+
)
6967

7068
# Check that all task IDs are present
7169
expected_task_ids = {f"test_task_{i}" for i in range(num_parallel_saves)}
7270
actual_task_ids = set(result_df["id"].values)
73-
assert (
74-
expected_task_ids == actual_task_ids
75-
), f"Missing task IDs: {expected_task_ids - actual_task_ids}"
71+
assert expected_task_ids == actual_task_ids, (
72+
f"Missing task IDs: {expected_task_ids - actual_task_ids}"
73+
)
7674

7775
# Check that all folds are present and unique
7876
expected_folds = set(range(num_parallel_saves))
7977
actual_folds = set(result_df["fold"].values)
80-
assert (
81-
expected_folds == actual_folds
82-
), f"Missing folds: {expected_folds - actual_folds}"
78+
assert expected_folds == actual_folds, (
79+
f"Missing folds: {expected_folds - actual_folds}"
80+
)
8381

8482

8583
def test_save_with_file_lock_timeout(mocker):

0 commit comments

Comments
 (0)