From c8f3dbe2b3cd8b2c0593bfbe748ff97a0d748172 Mon Sep 17 00:00:00 2001
From: mttga <matteogallici@yahoo.it>
Date: Wed, 7 Feb 2024 02:34:09 +0100
Subject: [PATCH] stupid typo when choosing the agent net

---
 baselines/QLearning/transf_qmix.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/baselines/QLearning/transf_qmix.py b/baselines/QLearning/transf_qmix.py
index 3a7d6d13..3d131bb9 100644
--- a/baselines/QLearning/transf_qmix.py
+++ b/baselines/QLearning/transf_qmix.py
@@ -426,7 +426,7 @@ def _env_sample_step(env_state, unused):
 
         # INIT NETWORK
         # init agent
-        if env.name=='smax': # smax agent 
+        if 'smax' in env.name.lower(): # smax agent 
             agent_class = TransformerAgentSmax
             n_entities = wrapped_env._env.num_allies+wrapped_env._env.num_enemies # must be explicit for the n_entities if using policy decoupling
             init_x = (