From dd9c6eed15efb28d82c5742c59d09560417cb65d Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 17 Apr 2024 16:27:57 +0100 Subject: [PATCH] Add temporary patch in TD-MPC --- lerobot/common/policies/tdmpc/policy.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lerobot/common/policies/tdmpc/policy.py b/lerobot/common/policies/tdmpc/policy.py index 14728576..ed28c4a6 100644 --- a/lerobot/common/policies/tdmpc/policy.py +++ b/lerobot/common/policies/tdmpc/policy.py @@ -330,6 +330,10 @@ class TDMPCPolicy(nn.Module): return td_target def forward(self, batch, step): + # TODO(alexander-soare): Refactor TDMPC and make it comply with the policy interface documentation. + raise NotImplementedError() + + def update(self, batch, step): """Main update function. Corresponds to one iteration of the model learning.""" start_time = time.time()