zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-19 07:31:26 +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 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,