fix: rec_qmix import

instadeepai · Nov 13, 2024 · 7da5968 · 7da5968
1 parent acf1830
commit 7da5968
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mava/systems/q_learning/anakin/rec_qmix.py b/mava/systems/q_learning/anakin/rec_qmix.py
@@ -47,13 +47,13 @@
 from mava.types import MarlEnv, Observation
 from mava.utils import make_env as environments
 from mava.utils.checkpointing import Checkpointer
+from mava.utils.config import check_total_timesteps
 from mava.utils.jax_utils import (
     switch_leading_axes,
     unreplicate_batch_dim,
     unreplicate_n_dims,
 )
 from mava.utils.logger import LogEvent, MavaLogger
-from mava.utils.total_timestep_checker import check_total_timesteps
 from mava.wrappers import episode_metrics