fix: incorrect condition control flow

#261
by tpob - opened
Files changed (1) hide show
  1. geneformer/pretrainer.py +1 -1
geneformer/pretrainer.py CHANGED
@@ -381,7 +381,7 @@ class GeneformerPreCollator(SpecialTokensMixin):
381
  return_tensors = "tf" if return_tensors is None else return_tensors
382
  elif is_torch_available() and _is_torch(first_element):
383
  return_tensors = "pt" if return_tensors is None else return_tensors
384
- if isinstance(first_element, np.ndarray):
385
  return_tensors = "np" if return_tensors is None else return_tensors
386
  else:
387
  raise ValueError(
 
381
  return_tensors = "tf" if return_tensors is None else return_tensors
382
  elif is_torch_available() and _is_torch(first_element):
383
  return_tensors = "pt" if return_tensors is None else return_tensors
384
+ elif isinstance(first_element, np.ndarray):
385
  return_tensors = "np" if return_tensors is None else return_tensors
386
  else:
387
  raise ValueError(