Potential redundant slicing

[emergenz]

Check whether we need the [:2] slicing in:

jasmine/jasmine/genie.py

Line 166 in b8c92e7

*batch["actions"].shape[:2], 1, self.latent_action_dim