fix fastmath for stump._compute_diagonal

NimaSarajpoor · NimaSarajpoor · commit 2d3a6ac14a62 · 2025-01-30T20:16:10.000-05:00
diff --git a/stumpy/stump.py b/stumpy/stump.py
@@ -15,7 +15,7 @@
     # "(f8[:], f8[:], i8, f8[:], f8[:], f8[:], f8[:], f8[:], f8[:], f8[:], f8[:],"
     # "b1[:], b1[:], b1[:], b1[:], i8[:], i8, i8, i8, f8[:, :, :], f8[:, :],"
     # "f8[:, :], i8[:, :, :], i8[:, :], i8[:, :], b1)",
-    fastmath=config.STUMPY_FASTMATH_TRUE,
+    fastmath=config.STUMPY_FASTMATH_FLAGS,
 )
 def _compute_diagonal(
     T_A,

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@`
`15`	`15`	`# "(f8[:], f8[:], i8, f8[:], f8[:], f8[:], f8[:], f8[:], f8[:], f8[:], f8[:],"`
`16`	`16`	`# "b1[:], b1[:], b1[:], b1[:], i8[:], i8, i8, i8, f8[:, :, :], f8[:, :],"`
`17`	`17`	`# "f8[:, :], i8[:, :, :], i8[:, :], i8[:, :], b1)",`
`18`		`- fastmath=config.STUMPY_FASTMATH_TRUE,`
	`18`	`+ fastmath=config.STUMPY_FASTMATH_FLAGS,`
`19`	`19`	`)`
`20`	`20`	`def _compute_diagonal(`
`21`	`21`	`T_A,`