From d79f48ffdfc1d706560b49f497a7648b48fdca25 Mon Sep 17 00:00:00 2001
From: Matt Fisher <mfisher87@gmail.com>
Date: Sat, 6 Jul 2024 13:02:07 -0600
Subject: [PATCH] Use union type instead of union operator

---
 tests/integration/sample.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/integration/sample.py b/tests/integration/sample.py
index cd8e1f5e..dc24df36 100644
--- a/tests/integration/sample.py
+++ b/tests/integration/sample.py
@@ -1,5 +1,6 @@
 import logging
 import random
+from typing import Union
 
 logger = logging.getLogger(__name__)
 
@@ -7,7 +8,7 @@
 def get_sample_granules(
     granules: list,
     sample_size: int,
-    max_granule_size: int | float,
+    max_granule_size: Union[int, float],
     round_ndigits: int = None,
 ):
     """Return a list of randomly-sampled granules and their size in MB.