Fix for Issue #123 (#178)

* Update torch_wrapper.py * Update torch_wrapper.py * Update torch_wrapper.py * remove redundant comparisons --------- Co-authored-by: Hideaki Takahashi <koukyosyumei@hotmail.com>
Koukyosyumei · Apr 15, 2024 · 4f4b8da · 4f4b8da
1 parent 54886fa
commit 4f4b8da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aijack/defense/paillier/torch_wrapper.py b/src/aijack/defense/paillier/torch_wrapper.py
@@ -21,9 +21,9 @@ class PaillierTensor(object):
     """torch.Tensor-like object for Paillier Encryption"""
 
     def __init__(self, paillier_array):
-        if type(paillier_array) == list:
+        if isinstance(paillier_array, list):
             self._paillier_np_array = np.array(paillier_array)
-        elif type(paillier_array) == np.ndarray:
+        elif isinstance(paillier_array, np.ndarray):
             self._paillier_np_array = paillier_array
         else:
             raise TypeError(f"{type(paillier_array)} is not supported.")