Skip to content

Commit c9ab33b

Browse files
committed
Lint and fix test
Signed-off-by: Ilya Sherstyuk <[email protected]>
1 parent e6deafb commit c9ab33b

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

collector/helper.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
import signal
1111
import sys
1212
import traceback
13-
1413
from datetime import datetime
1514
from pathlib import Path
1615

src/aiconfigurator/sdk/perf_database.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -348,7 +348,7 @@ def load_moe_data(moe_file):
348348
"""
349349
if not os.path.exists(moe_file):
350350
logger.warning(f"MOE data file {moe_file} not found.")
351-
return None
351+
return None, None
352352

353353
moe_default_data = defaultdict(
354354
lambda: defaultdict(

tests/sdk/database/test_data_loaders.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def test_load_moe_data_nonexistent(tmp_path):
253253
"""
254254
fake_path = tmp_path / "no_moe.csv"
255255
result = load_moe_data(str(fake_path))
256-
assert result is None
256+
assert result == (None, None)
257257

258258

259259
def test_load_moe_data_basic(tmp_path):

0 commit comments

Comments
 (0)