zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 01:17:26 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix style

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-06-20 16:45:00 -07:00
parent 90072445e7
commit 8ec7713893
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_nn.py
View File

@ -260,7 +260,7 @@ class TestBase(mlx_tests.MLXTestCase):
m = m.update_modules({"list": ["hi"]}) m = m.update_modules({"list": ["hi"]})
# Allow updating a strict subset # Allow updating a strict subset
m = nn.Sequential(nn.Linear(3, 3), nn. Linear(3, 3)) m = nn.Sequential(nn.Linear(3, 3), nn.Linear(3, 3))
m.update_modules({"layers": [{}, nn.Linear(3, 4)]}) m.update_modules({"layers": [{}, nn.Linear(3, 4)]})
self.assertEqual(m.layers[1].weight.shape, (4, 3)) self.assertEqual(m.layers[1].weight.shape, (4, 3))