From 69da561f442e137defeb8d0d68af965d99d88e2a Mon Sep 17 00:00:00 2001 From: David Carreto Fidalgo Date: Mon, 9 Dec 2024 20:27:31 +0100 Subject: [PATCH] fix asha test --- tests/test_asha.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_asha.py b/tests/test_asha.py index f602ef6..65df59b 100644 --- a/tests/test_asha.py +++ b/tests/test_asha.py @@ -1,8 +1,7 @@ import pandas as pd import pytest -from slurm_sweeps import ASHA -from slurm_sweeps.asha import AshaConfig +from slurm_sweeps.asha import ASHA, AshaConfig from slurm_sweeps.constants import DB_ITERATION, DB_METRIC, DB_TRIAL_ID