Fix the reduce method for Ivy Shape (#28212)

ivy-llc · Feb 9, 2024 · 81a0371 · 81a0371
1 parent c60b3a2
commit 81a0371
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/__init__.py b/ivy/__init__.py
@@ -312,7 +312,7 @@ def __rmod__(self, other):
         return other % self._shape
 
     def __reduce__(self):
-        return (self._shape,)
+        return (self.__class__, (self._shape,))
 
     def as_dimension(self, other):
         if isinstance(other, self._shape):