From fdfb2010fd4bde9278431602f4f3466c8f284342 Mon Sep 17 00:00:00 2001 From: Cadene Date: Sun, 18 Feb 2024 01:24:19 +0000 Subject: [PATCH] black --- lerobot/common/tdmpc.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/lerobot/common/tdmpc.py b/lerobot/common/tdmpc.py index 03aeb7d7..de4543db 100644 --- a/lerobot/common/tdmpc.py +++ b/lerobot/common/tdmpc.py @@ -128,10 +128,7 @@ class TDMPC(nn.Module): def act(self, obs, t0=False, step=None): """Take an action. Uses either MPC or the learned policy, depending on the self.cfg.mpc flag.""" if isinstance(obs, dict): - obs = { - k: o.detach().unsqueeze(0) - for k, o in obs.items() - } + obs = {k: o.detach().unsqueeze(0) for k, o in obs.items()} else: obs = obs.detach().unsqueeze(0) z = self.model.encode(obs)