[SNOW-1458256]: Ensure binary operations are present in telemetry (#1732

)
snowflakedb · Jun 4, 2024 · 11ea0a4 · 11ea0a4
1 parent e7813c1
commit 11ea0a4
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/snowflake/snowpark/modin/plugin/_internal/telemetry.py b/src/snowflake/snowpark/modin/plugin/_internal/telemetry.py
@@ -470,6 +470,28 @@ def wrap(*args, **kwargs):  # type: ignore
     "__setitem__",
     "__iter__",
     "__repr__",
+    "__add__",
+    "__iadd__",
+    "__radd__",
+    "__mul__",
+    "__imul__",
+    "__rmul__",
+    "__pow__",
+    "__ipow__",
+    "__rpow__",
+    "__sub__",
+    "__isub__",
+    "__rsub__",
+    "__floordiv__",
+    "__ifloordiv__",
+    "__rfloordiv__",
+    "__truediv__",
+    "__itruediv__",
+    "__rtruediv__",
+    "__mod__",
+    "__imod__",
+    "__rmod__",
+    "__rdiv__",
 }