zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 16:48:10 +08:00
Code Issues Actions 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

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):