isinstance(x, int) -> isinstance(x, (int, np.integer)) (#627)

* isinstance(x, int) -> isinstance(x, (int, np.integer))
This commit is contained in:
Xueqing Liu
2022-07-06 13:22:05 -04:00
committed by GitHub
parent 6108493e0b
commit 6cb6a2a19a

View File

@@ -562,7 +562,7 @@ class LabelEncoderforTokenClassification:
# if the labels are not tokens, they must be ids
else:
assert all(
isinstance(id, int) for id in y[0]
isinstance(id, (int, np.integer)) for id in y[0]
), "The labels must either be tokens or ids"
return y