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()