(fix)(stateful)(module.py): fixed a bug within the build method for `…

…_KerasIvyModule`
ivy-llc · Jun 2, 2024 · e36b7a9 · e36b7a9
1 parent 9a5a578
commit e36b7a9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ivy/stateful/module.py b/ivy/stateful/module.py
@@ -777,7 +777,10 @@ def _create_variables(self, device=None, dtype=None):
     def _build(self, *args, **kwargs):
         self._native_params = ivy.Container(
             OrderedDict(
-                sorted([(param.name, param) for param in self._native_module.variables])
+                sorted(
+                    [(param.name, param) for param in self._native_module.variables],
+                    key=lambda kv: kv[0],
+                )
             ),
             dynamic_backend=False,
         )