(
self,
states,
actions,
goals,
sigma
)
| 420 | return pred_actions |
| 421 | |
| 422 | def forward( |
| 423 | self, |
| 424 | states, |
| 425 | actions, |
| 426 | goals, |
| 427 | sigma |
| 428 | ): |
| 429 | |
| 430 | if self.forward_type == 'cross_attn': |
| 431 | return self.cross_attn_forward(states, actions, goals, sigma) |
| 432 | elif self.forward_type == 'context_token': |
| 433 | return self.context_token_forward(states, actions, goals, sigma) |
| 434 | else: |
| 435 | raise ValueError(f"Invalid forward type: {self.forward_type}") |
| 436 |
nothing calls this directly
no test coverage detected