test: update loss probe helper call

MilesCranmerBot · MilesCranmer · MilesCranmerBot · commit 054de3765d70 · 2026-04-18T16:13:37.000Z
Co-authored-by: Miles Cranmer &lt;miles.cranmer@gmail.com&gt;
diff --git a/pysr/test/test_main.py b/pysr/test/test_main.py
@@ -320,7 +320,7 @@ def test_elementwise_loss_float32_probe_accepts_strictly_typed_loss(self):
         _validate_elementwise_loss(
             custom_loss,
             has_weights=False,
-            probe_dtype=np.float32,
+            probe_value=np.float32(1.0),
         )
 
     def test_elementwise_loss_float32_fit_accepts_strictly_typed_loss(self):

Original file line number	Diff line number	Diff line change
`@@ -320,7 +320,7 @@ def test_elementwise_loss_float32_probe_accepts_strictly_typed_loss(self):`
`320`	`320`	`_validate_elementwise_loss(`
`321`	`321`	`custom_loss,`
`322`	`322`	`has_weights=False,`
`323`		`- probe_dtype=np.float32,`
	`323`	`+ probe_value=np.float32(1.0),`
`324`	`324`	`)`
`325`	`325`
`326`	`326`	`def test_elementwise_loss_float32_fit_accepts_strictly_typed_loss(self):`