jax
7241ee09 - [Pallas/Mosaic GPU] Use `self.capture_stdout` instead of `jtu.capture_stdout` in `mosaic_gpu_test`.

Commit

27 days ago

[Pallas/Mosaic GPU] Use `self.capture_stdout` instead of `jtu.capture_stdout` in `mosaic_gpu_test`. This correctly skips the test under pytest. PiperOrigin-RevId: 861723159

References

#34652 - [Pallas/Mosaic GPU] Use `self.capture_stdout` instead of `jtu.capture_stdout` in `mosaic_gpu_test`.

Author

bchetioui

Committer

Google-ML-Automation

Parents

9663c503

jax 7241ee09 - [Pallas/Mosaic GPU] Use `self.capture_stdout` instead of `jtu.capture_stdout` in `mosaic_gpu_test`.

jax
7241ee09 - [Pallas/Mosaic GPU] Use `self.capture_stdout` instead of `jtu.capture_stdout` in `mosaic_gpu_test`.