From 378c61caa1cab824c66a6c7982b9d25d78715836 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Thu, 21 Sep 2023 14:50:36 +0100 Subject: [PATCH] update models Signed-off-by: Matteo Bettini --- benchmarl/models/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarl/models/common.py b/benchmarl/models/common.py index 6ce0fc61..d4c7c03a 100644 --- 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)}" )