Unpin numba (#23162)
* fix for ragged list * unpin numba * make style * np.object -> object * propagate changes to tokenizer as well * np.long -> "long" * revert tokenization changes * check with tokenization changes * list/tuple logic * catch numpy * catch else case * clean up * up * better check * trigger ci * Empty commit to trigger CI
This commit is contained in:
@@ -392,7 +392,7 @@ class RealmModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
|
||||
b"This is the fourth record.",
|
||||
b"This is the fifth record.",
|
||||
],
|
||||
dtype=np.object,
|
||||
dtype=object,
|
||||
)
|
||||
retriever = RealmRetriever(block_records, tokenizer)
|
||||
model = RealmForOpenQA(openqa_config, retriever)
|
||||
|
||||
@@ -100,7 +100,7 @@ class RealmRetrieverTest(TestCase):
|
||||
b"This is the fifth record",
|
||||
b"This is a longer longer longer record",
|
||||
],
|
||||
dtype=np.object,
|
||||
dtype=object,
|
||||
)
|
||||
return block_records
|
||||
|
||||
@@ -116,7 +116,7 @@ class RealmRetrieverTest(TestCase):
|
||||
retriever = self.get_dummy_retriever()
|
||||
tokenizer = retriever.tokenizer
|
||||
|
||||
retrieved_block_ids = np.array([0, 3], dtype=np.long)
|
||||
retrieved_block_ids = np.array([0, 3], dtype="long")
|
||||
question_input_ids = tokenizer(["Test question"]).input_ids
|
||||
answer_ids = tokenizer(
|
||||
["the fourth"],
|
||||
@@ -151,7 +151,7 @@ class RealmRetrieverTest(TestCase):
|
||||
retriever = self.get_dummy_retriever()
|
||||
tokenizer = retriever.tokenizer
|
||||
|
||||
retrieved_block_ids = np.array([0, 3, 5], dtype=np.long)
|
||||
retrieved_block_ids = np.array([0, 3, 5], dtype="long")
|
||||
question_input_ids = tokenizer(["Test question"]).input_ids
|
||||
answer_ids = tokenizer(
|
||||
["the fourth", "longer longer"],
|
||||
|
||||
Reference in New Issue
Block a user