amend

facebookresearch · Sep 21, 2024 · 45a9395 · 45a9395
1 parent 3d2174d
commit 45a9395
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/benchmarl/models/gnn.py b/benchmarl/models/gnn.py
@@ -287,7 +287,6 @@ def _forward(self, tensordict: TensorDictBase) -> TensorDictBase:
                         f"Position key in tensordict is {pos.shape[-1]}-dimensional, "
                         f"while model was configured with pos_features={self.pos_features-1}"
                     )
-                print(self._full_position_key)
             else:
                 pos = tensordict.get(self._full_position_key)
             if not self.exclude_pos_from_node_features: