remove environment state conditioning

This commit is contained in:
Bryson Jones
2025-12-11 08:26:21 -08:00
parent ba968e84f1
commit 86e0ee787d

View File

@@ -755,9 +755,6 @@ class MultiTaskDiTPolicy(PreTrainedPolicy):
if self.config.image_features:
self._queues["observation.images"] = deque(maxlen=self.config.n_obs_steps)
if self.config.env_state_feature:
self._queues["observation.environment_state"] = deque(maxlen=self.config.n_obs_steps)
self._queues["task"] = deque(maxlen=self.config.n_obs_steps)
@torch.no_grad()