backup wip

2024-03-19 16:02:09 +00:00
parent 88347965c2
commit ea17f4ce50
11 changed files with 71 additions and 46 deletions
--- a/lerobot/common/policies/abstract.py
+++ b/lerobot/common/policies/abstract.py
@@ -30,7 +30,7 @@ class AbstractPolicy(nn.Module, ABC):
        Should return a (batch_size, n_action_steps, *) tensor of actions.
        """

-    def forward(self, *args, **kwargs):
+    def forward(self, *args, **kwargs) -> Tensor:
        """Inference step that makes multi-step policies compatible with their single-step environments.

        WARNING: In general, this should not be overriden.