Fixed weird type checker error

patrick-kidger · patrick-kidger · Jun 29, 2024 · Feb 21, 2024 · May 7, 2024 · May 15, 2024
commit eb16e57c8e521f5915d6ee98515c5b3d1680a804
diff --git a/diffrax/_integrate.py b/diffrax/_integrate.py
@@ -553,7 +553,7 @@ def _outer_cond_fn(cond_fn_i, old_event_value_i):
             event_structure = jtu.tree_structure(event.cond_fn, is_leaf=callable)
             event_values, event_mask = jtu.tree_transpose(
                 event_structure,
-                None,  # pyright: ignore
+                jtu.tree_structure((0, 0)),
                 event_values__mask,
             )
             had_event = False
@@ -1334,7 +1334,7 @@ def _outer_cond_fn(cond_fn_i):
         event_structure = jtu.tree_structure(event.cond_fn, is_leaf=callable)
         event_values, event_mask = jtu.tree_transpose(
             event_structure,
-            None,  # pyright: ignore
+            jtu.tree_structure((0, 0)),
             event_values__mask,
         )
         had_event = False