Adjust TensorArray.isna() to avoid a corner case

CODAIT · Jul 7, 2021 · 777ac2f · 777ac2f
1 parent 50a985e
commit 777ac2f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/text_extensions_for_pandas/array/tensor.py b/text_extensions_for_pandas/array/tensor.py
@@ -349,7 +349,12 @@ def isna(self) -> np.array:
  for information about this method.
  """
  if self._tensor.dtype.type is np.object_:
- return self._tensor == None
+ # Avoid comparing with __eq__ because the elements of the tensor may do
+ # something funny with that operation.
+ result_list = [
+ self._tensor[i] is None for i in range(len(self))
+ ]
+ return np.array(result_list, dtype=bool)
  elif self._tensor.dtype.type is np.str_:
  return np.all(self._tensor == "", axis=-1)
  else: