to syntax

sammccallum · sammccallum · commit 8d59058b82f5 · 2025-10-12T16:03:51.000+01:00
diff --git a/diffrax/_integrate.py b/diffrax/_integrate.py
@@ -115,9 +115,9 @@ class State(eqx.Module):
     #
     # Information for reversible adjoint (save ts)
     #
-    reversible_init_ts: Optional[PyTree[FloatScalarLike]]
-    reversible_ts: Optional[eqxi.MaybeBuffer[Float[Array, " times_plus_1"]]]
-    reversible_save_index: Optional[IntScalarLike]
+    reversible_init_ts: PyTree[FloatScalarLike] | None
+    reversible_ts: eqxi.MaybeBuffer[Float[Array, " times_plus_1"]] | None
+    reversible_save_index: IntScalarLike | None
 
 
 def _is_none(x: Any) -> bool:

Original file line number	Diff line number	Diff line change
`@@ -115,9 +115,9 @@ class State(eqx.Module):`
`115`	`115`	`#`
`116`	`116`	`# Information for reversible adjoint (save ts)`
`117`	`117`	`#`
`118`		`- reversible_init_ts: Optional[PyTree[FloatScalarLike]]`
`119`		`- reversible_ts: Optional[eqxi.MaybeBuffer[Float[Array, " times_plus_1"]]]`
`120`		`- reversible_save_index: Optional[IntScalarLike]`
	`118`	`+ reversible_init_ts: PyTree[FloatScalarLike] \| None`
	`119`	`+ reversible_ts: eqxi.MaybeBuffer[Float[Array, " times_plus_1"]] \| None`
	`120`	`+ reversible_save_index: IntScalarLike \| None`
`121`	`121`
`122`	`122`
`123`	`123`	`def _is_none(x: Any) -> bool:`