bugs

facebookresearch · Nov 5, 2024 · 4b9992e · 4b9992e
1 parent 39884c8
commit 4b9992e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -366,9 +366,9 @@ def _perfrom_checks(self):
         for config in model_configs:
             if isinstance(config, SequenceModelConfig):
                 for layer_config in self.critic_model_config.model_configs[1:]:
-                    if (
-                        isinstance(layer_config, GnnConfig)
-                        and layer_config.topology == "from_pos"
+                    if isinstance(layer_config, GnnConfig) and (
+                        layer_config.position_key is not None
+                        or layer_config.velocity_key is not None
                     ):
                         raise ValueError(
                             "GNNs with topology 'from_pos' are currently only usable in first"