diff --git a/.gitignore b/.gitignore index 176aa2af..f35eac8a 100644 --- a/.gitignore +++ b/.gitignore @@ -9,3 +9,4 @@ site/ .idea/ .venv/ uv.lock +.vscode/ \ No newline at end of file diff --git a/diffrax/_custom_types.py b/diffrax/_custom_types.py index eef13b12..24adc2e4 100644 --- a/diffrax/_custom_types.py +++ b/diffrax/_custom_types.py @@ -133,4 +133,4 @@ def levy_tree_transpose( ) -del Array, ArrayLike, PyTree, Bool, Int, Shaped, Float, AbstractDtype +del Array, ArrayLike, PyTree, Bool, Int, Shaped, Float, AbstractDtype, Real