Merge pull request #460 from rapidsai/branch-0.17

[gpuCI] Auto-merge branch-0.17 to branch-0.18 [skip ci]

dask_cuda/proxy_object.py

@@ -300,6 +300,24 @@ def __iter__(self):
     def __array__(self):
         return getattr(self._obj_pxy_deserialize(), "__array__")()
 
+    def __lt__(self, other):
+        return self._obj_pxy_deserialize() < other
+
+    def __le__(self, other):
+        return self._obj_pxy_deserialize() <= other
+
+    def __eq__(self, other):
+        return self._obj_pxy_deserialize() == other
+
+    def __ne__(self, other):
+        return self._obj_pxy_deserialize() != other
+
+    def __gt__(self, other):
+        return self._obj_pxy_deserialize() > other
+
+    def __ge__(self, other):
+        return self._obj_pxy_deserialize() >= other
+
     def __add__(self, other):
         return self._obj_pxy_deserialize() + other
 

dask_cuda/tests/test_proxy.py

@@ -134,6 +134,12 @@ def test_proxy_object_of_numpy(serializers):
         assert isinstance(got, type(expect))
         assert all(expect == got)
 
+        # Check proxy-proxy operations
+        if "i" != op_str[0]:  # Skip in-place operators
+            expect = op(org.copy(), org)
+            got = op(pxy, proxy_object.asproxy(org.copy()))
+            assert all(expect == got)
+
     # Check unary truth operators
     for op_str in ["not_", "truth"]:
         op = getattr(operator, op_str)