def one_step_on_data(data):
            """Runs a single test step on a batch of data."""
            data = data[0]
            with torch.no_grad():
                return self.test_step(data)
Runs a single test step on a batch of data.
keras/src/backend/torch/trainer.py