fix check (#8791)

Signed-off-by: Rahul <rahulhuilgol@gmail.com>

python/mxnet/kvstore.py

@@ -408,7 +408,7 @@ def set_gradient_compression(self, compression_params):
             Other keys in this dictionary are optional and specific to the type
             of gradient compression.
         """
-        if (self.type == 'device') or ('dist' in self.type):
+        if ('device' in self.type) or ('dist' in self.type):
             ckeys, cvals = _ctype_dict(compression_params)
             check_call(_LIB.MXKVStoreSetGradientCompression(self.handle,
                                                             mx_uint(len(compression_params)),