........................................................................ [ 56%]
......................................................s                  [100%]
=============================== warnings summary ===============================
tests/test_non_finite_logits_and_lazy_sampler_law.py::test_sample_from_logits_raises_on_every_non_finite_shape[0.6-row0]
tests/test_non_finite_logits_and_lazy_sampler_law.py::test_sample_from_logits_raises_on_every_non_finite_shape[1.0-row0]
  /tmp/ds4-issue103/mtplx/mtplx/fast_sampling.py:482: RuntimeWarning: All-NaN slice encountered
    spill = np.nanmin(cand_val_rows, axis=1) >= cutoff

tests/test_non_finite_logits_and_lazy_sampler_law.py::test_sample_from_logits_raises_on_every_non_finite_shape[0.6-row1]
tests/test_non_finite_logits_and_lazy_sampler_law.py::test_sample_from_logits_raises_on_every_non_finite_shape[0.6-row3]
tests/test_non_finite_logits_and_lazy_sampler_law.py::test_sample_from_logits_raises_on_every_non_finite_shape[1.0-row1]
tests/test_non_finite_logits_and_lazy_sampler_law.py::test_sample_from_logits_raises_on_every_non_finite_shape[1.0-row3]
  /tmp/ds4-issue103/mtplx/mtplx/sampling.py:138: RuntimeWarning: invalid value encountered in subtract
    scaled = scaled - np.max(scaled)

-- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html