From 0d31128a44e6c3f6c06607f932cc8e82f224e051 Mon Sep 17 00:00:00 2001
From: Diogo <dgcruz983@gmail.com>
Date: Wed, 3 Jan 2024 22:33:19 -0500
Subject: [PATCH] use union instead of | (#358)

---
 python/tests/mlx_tests.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/tests/mlx_tests.py b/python/tests/mlx_tests.py
index dfbc835da..01ef407c3 100644
--- a/python/tests/mlx_tests.py
+++ b/python/tests/mlx_tests.py
@@ -2,7 +2,7 @@
 
 import os
 import unittest
-from typing import Any, Callable, List, Tuple
+from typing import Any, Callable, List, Tuple, Union
 
 import mlx.core as mx
 import numpy as np
@@ -21,7 +21,7 @@ class MLXTestCase(unittest.TestCase):
 
     def assertCmpNumpy(
         self,
-        shape: List[Tuple[int] | Any],
+        shape: List[Union[Tuple[int], Any]],
         mx_fn: Callable[..., mx.array],
         np_fn: Callable[..., np.array],
         atol=1e-2,