update models

Signed-off-by: Matteo Bettini <[email protected]>
facebookresearch · Sep 21, 2023 · 378c61c · 378c61c
1 parent c82d964
commit 378c61c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarl/models/common.py b/benchmarl/models/common.py
@@ -207,7 +207,7 @@ def get_model(
             raise ValueError(
                 f"SequenceModelConfig expects n_models > 0, got {n_models}"
             )
-        if len(self.intermediate_sizes) == n_models - 1:
+        if len(self.intermediate_sizes) != n_models - 1:
             raise ValueError(
                 f"SequenceModelConfig intermediate_sizes len should be {n_models - 1}, got {len(self.intermediate_sizes)}"
             )