From 3301769e0b162b06e39c96390a98b78cffab8eb6 Mon Sep 17 00:00:00 2001 From: Manish Aradwad Date: Mon, 1 Jan 2024 17:23:10 +0000 Subject: [PATCH] precommit changes --- python/mlx/nn/layers/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py index 7c4ccc85e..c22d0adc2 100644 --- a/python/mlx/nn/layers/base.py +++ b/python/mlx/nn/layers/base.py @@ -444,6 +444,6 @@ class Module(dict): state = self.__dict__.copy() state.pop("_compiled_call_impl", None) return state - + def __setstate__(self, state): - self.__dict__.update(state) \ No newline at end of file + self.__dict__.update(state)