From 7da596853ce40c8173d759270752d530f7f4d2ad Mon Sep 17 00:00:00 2001 From: Louay Ben Nessir Date: Wed, 13 Nov 2024 13:05:34 +0100 Subject: [PATCH] fix: rec_qmix import --- mava/systems/q_learning/anakin/rec_qmix.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mava/systems/q_learning/anakin/rec_qmix.py b/mava/systems/q_learning/anakin/rec_qmix.py index 2b485bd09..7dcccf75c 100644 --- 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