zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-21 16:51:15 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

use union instead of | (#358)

Browse Source
This commit is contained in:
Diogo 2024-01-03 22:33:19 -05:00 committed by GitHub
parent 1ac18eac20
commit 0d31128a44
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/tests/mlx_tests.py
View File

@ -2,7 +2,7 @@
import os import os
import unittest import unittest
from typing import Any, Callable, List, Tuple from typing import Any, Callable, List, Tuple, Union
import mlx.core as mx import mlx.core as mx
import numpy as np import numpy as np
@ -21,7 +21,7 @@ class MLXTestCase(unittest.TestCase):
def assertCmpNumpy( def assertCmpNumpy(
self, self,
shape: List[Tuple[int] | Any], shape: List[Union[Tuple[int], Any]],
mx_fn: Callable[..., mx.array], mx_fn: Callable[..., mx.array],
np_fn: Callable[..., np.array], np_fn: Callable[..., np.array],
atol=1e-2, atol=1e-2,