From f3219598b95b003002241338635563279894ea1b Mon Sep 17 00:00:00 2001
From: Pablo Zubieta <8410335+p-zubieta@users.noreply.github.com>
Date: Thu, 15 Jan 2026 09:26:27 -0600
Subject: [PATCH] Import Union from pysages.typing, not from plum

---
 pysages/grids.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pysages/grids.py b/pysages/grids.py
index 6c2039ee..8be7987e 100644
--- a/pysages/grids.py
+++ b/pysages/grids.py
@@ -5,9 +5,9 @@
 
 from jax import jit
 from jax import numpy as np
-from plum import Union, parametric
+from plum import parametric
 
-from pysages.typing import JaxArray
+from pysages.typing import JaxArray, Union
 from pysages.utils import dispatch, is_generic_subclass, prod