Flaky test sharktank\tests\types\dataset_test.py::ThetaTest::testTransform

[ScottTodd]

NaNs?

• https://github.com/nod-ai/sharktank/actions/runs/9715748619/job/26817850504
• https://github.com/nod-ai/sharktank/actions/runs/9671705133/job/26682786147
• https://github.com/nod-ai/sharktank/actions/runs/9668291276/job/26672106534
• https://github.com/nod-ai/sharktank/actions/runs/9663612655/job/26656199637

FAILED sharktank\tests\types\dataset_test.py::ThetaTest::testTransform - AssertionError: Tensor-likes are not close!

Mismatched elements: 1 / 2 (50.0%)
Greatest absolute difference: nan at index (0, 1) (up to 1e-05 allowed)
Greatest relative difference: nan at index (0, 1) (up to 1.3e-06 allowed)