Add float32 to test -- manually :O

calebrob6 · web-flow · commit 4aae9885c3de · 2026-04-29T23:47:05.000-07:00
diff --git a/tests/test_dtype_consistency.py b/tests/test_dtype_consistency.py
@@ -48,7 +48,7 @@ def _make_sample(dtype: torch.dtype) -> MaskedOlmoEarthSample:
     )
 
 
-@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16])
+@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32])
 def test_nano_encoder_forward_low_precision(dtype: torch.dtype) -> None:
     """Encoder forward pass must not raise a dtype mismatch in fp16 / bf16."""
     model = OlmoEarthPretrain_v1(

Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ def _make_sample(dtype: torch.dtype) -> MaskedOlmoEarthSample:`
`48`	`48`	`)`
`49`	`49`
`50`	`50`
`51`		`-@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16])`
	`51`	`+@pytest.mark.parametrize("dtype", [torch.float16, torch.bfloat16, torch.float32])`
`52`	`52`	`def test_nano_encoder_forward_low_precision(dtype: torch.dtype) -> None:`
`53`	`53`	`"""Encoder forward pass must not raise a dtype mismatch in fp16 / bf16."""`
`54`	`54`	`model = OlmoEarthPretrain_v1(`