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

revert copy (#366)

Browse Source
This commit is contained in:
Awni Hannun
2024-01-04 10:43:29 -08:00
committed by GitHub
parent 16856a0160
commit cf88db44b5
3 changed files with 0 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/mlx/nn/layers/base.py
View File

@@ -530,9 +530,3 @@ class Module(dict):
See :func:`train`.
"""
self.train(False)
def __getstate__(self):
return self.__dict__.copy()
def __setstate__(self, state):
self.__dict__.update(state)

13
python/tests/test_nn.py
View File

@@ -854,19 +854,6 @@ class TestLayers(mlx_tests.MLXTestCase):
self.assertTrue(y.shape, x.shape)
self.assertTrue(y.dtype, mx.float16)
def test_deepcopy(self):
import copy
layer = nn.Linear(input_dims=4, output_dims=8)
layer_copy = copy.deepcopy(layer)
# Verify that the copied layer is not the same object as the original layer
self.assertIsNot(layer_copy, layer)
# Verify that the copied layer has the same attributes as the original layer
self.assertEqual(layer_copy.input_dims, layer.input_dims)
self.assertEqual(layer_copy.output_dims, layer.output_dims)
if __name__ == "__main__":
unittest.main()