Fixed ordering of right side reversed operators

ndif-team · Jan 3, 2024 · 8e91538 · 8e91538
1 parent d8af49b
commit 8e91538
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/nnsight/tracing/Proxy.py b/src/nnsight/tracing/Proxy.py
@@ -103,7 +103,7 @@ def __add__(self, other: Union[Proxy, Any]) -> Proxy:
     def __radd__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.add,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __sub__(self, other: Union[Proxy, Any]) -> Proxy:
@@ -115,7 +115,7 @@ def __sub__(self, other: Union[Proxy, Any]) -> Proxy:
     def __rsub__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.sub,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __pow__(self, other: Union[Proxy, Any]) -> Proxy:
@@ -127,7 +127,7 @@ def __pow__(self, other: Union[Proxy, Any]) -> Proxy:
     def __rpow__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.pow,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __mul__(self, other: Union[Proxy, Any]) -> Proxy:
@@ -139,7 +139,7 @@ def __mul__(self, other: Union[Proxy, Any]) -> Proxy:
     def __rmul__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.mul,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __mod__(self, other: Union[Proxy, Any]) -> Proxy:
@@ -151,7 +151,7 @@ def __mod__(self, other: Union[Proxy, Any]) -> Proxy:
     def __rmod__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.mod,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __matmul__(self, other: Union[Proxy, Any]) -> Proxy:
@@ -163,7 +163,7 @@ def __matmul__(self, other: Union[Proxy, Any]) -> Proxy:
     def __rmatmul__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.matmul,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __truediv__(self, other: Union[Proxy, Any]) -> Proxy:
@@ -175,7 +175,7 @@ def __truediv__(self, other: Union[Proxy, Any]) -> Proxy:
     def __rtruediv__(self, other: Union[Proxy, Any]) -> Proxy:
         return self.node.graph.add(
             target=operator.truediv,
-            args=[self.node, other],
+            args=[other, self.node],
         )
 
     def __bool__(self) -> bool: