zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-10 14:09:19 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add isort pre-commit and run (#68)

Browse Source
This commit is contained in:
Zach Schillaci
2023-12-08 20:31:47 +01:00
committed by GitHub
parent e89c571de7
commit 5b9be57ac3
33 changed files with 54 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/tests/test_array.py
View File

@@ -5,9 +5,8 @@ import unittest
from itertools import permutations
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
class TestVersion(mlx_tests.MLXTestCase):

1
python/tests/test_autograd.py
View File

@@ -3,7 +3,6 @@
import unittest
import mlx.core as mx
import mlx_tests

5
python/tests/test_bf16.py
View File

@@ -1,13 +1,12 @@
# Copyright © 2023 Apple Inc.
import math
import unittest
from itertools import permutations
import math
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
try:
import torch

5
python/tests/test_blas.py
View File

@@ -1,13 +1,12 @@
# Copyright © 2023 Apple Inc.
import math
import unittest
from itertools import permutations
import math
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
class TestBlas(mlx_tests.MLXTestCase):

5
python/tests/test_conv.py
View File

@@ -1,13 +1,12 @@
# Copyright © 2023 Apple Inc.
import math
import unittest
from itertools import permutations
import math
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
try:
import torch

1
python/tests/test_device.py
View File

@@ -3,7 +3,6 @@
import unittest
import mlx.core as mx
import mlx_tests

4
python/tests/test_eval.py
View File

@@ -1,11 +1,9 @@
# Copyright © 2023 Apple Inc.
import unittest
from functools import partial
import unittest
import mlx.core as mx
import mlx_tests

5
python/tests/test_fft.py
View File

@@ -1,12 +1,11 @@
# Copyright © 2023 Apple Inc.
import itertools
import unittest
import itertools
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
class TestFFT(mlx_tests.MLXTestCase):

6
python/tests/test_load.py
View File

@@ -1,12 +1,12 @@
# Copyright © 2023 Apple Inc.
import unittest
import os
import mlx.core as mx
import numpy as np
import tempfile
import unittest
import mlx.core as mx
import mlx_tests
import numpy as np
class TestLoad(mlx_tests.MLXTestCase):

9
python/tests/test_nn.py
View File

@@ -1,15 +1,14 @@
# Copyright © 2023 Apple Inc.
import os
import tempfile
import unittest
import mlx.core as mx
import mlx.nn as nn
from mlx.utils import tree_flatten, tree_map, tree_unflatten
import numpy as np
import os
import tempfile
import mlx_tests
import numpy as np
from mlx.utils import tree_flatten, tree_map, tree_unflatten
class TestNN(mlx_tests.MLXTestCase):

5
python/tests/test_ops.py
View File

@@ -1,13 +1,12 @@
# Copyright © 2023 Apple Inc.
import math
import unittest
from itertools import permutations
import math
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
class TestOps(mlx_tests.MLXTestCase):

1
python/tests/test_optimizers.py
View File

@@ -5,7 +5,6 @@ import unittest
import mlx.core as mx
import mlx.optimizers as opt
import mlx.utils
import mlx_tests

1
python/tests/test_random.py
View File

@@ -3,7 +3,6 @@
import unittest
import mlx.core as mx
import mlx_tests

5
python/tests/test_reduce.py
View File

@@ -1,12 +1,11 @@
# Copyright © 2023 Apple Inc.
import unittest
from itertools import permutations, combinations
from itertools import combinations, permutations
import mlx.core as mx
import numpy as np
import mlx_tests
import numpy as np
class TestReduce(mlx_tests.MLXTestCase):

1
python/tests/test_tree.py
View File

@@ -4,7 +4,6 @@ import unittest
import mlx.core as mx
import mlx.utils
import mlx_tests

1
python/tests/test_vmap.py
View File

@@ -3,7 +3,6 @@
import unittest
import mlx.core as mx
import mlx_tests