Skip to content

Commit

Permalink
make rag tests smaller (#10679)
Browse files Browse the repository at this point in the history
  • Loading branch information
@patrickvonplaten
patrickvonplaten committed Mar 15, 2021
1 parent 4c32f9f commit bd8f6ca
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 56 deletions.
28 changes: 0 additions & 28 deletions tests/test_modeling_rag.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -840,13 +840,6 @@ def test_data_questions(self):
"when is the last time the philadelphia won the superbowl",
"what is the most current adobe flash player version",
"how many episodes are there in dragon ball z",
"what is the first step in the evolution of the eye",
"where is gall bladder situated in human body",
"what is the main mineral in lithium batteries",
"who is the president of usa right now",
"where do the greasers live in the outsiders",
"panda is a national animal of which country",
"what is the name of manchester united stadium",
]

@slow
Expand Down Expand Up @@ -885,13 +878,6 @@ def test_rag_sequence_generate_batch(self):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

Expand Down Expand Up @@ -942,13 +928,6 @@ def test_rag_sequence_generate_batch_from_context_input_ids(self):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

Expand Down Expand Up @@ -986,13 +965,6 @@ def test_rag_token_generate_batch(self):
" the 1970s",
" 7.1. 2",
" 13",
" step by step",
" stomach",
" spodumene",
" obama",
" northern new jersey",
" india",
" united stadium",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

Expand Down
28 changes: 0 additions & 28 deletions tests/test_modeling_tf_rag.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -786,13 +786,6 @@ def test_data_questions(self):
"when is the last time the philadelphia won the superbowl",
"what is the most current adobe flash player version",
"how many episodes are there in dragon ball z",
"what is the first step in the evolution of the eye",
"where is gall bladder situated in human body",
"what is the main mineral in lithium batteries",
"who is the president of usa right now",
"where do the greasers live in the outsiders",
"panda is a national animal of which country",
"what is the name of manchester united stadium",
]

@slow
Expand Down Expand Up @@ -861,13 +854,6 @@ def test_rag_token_generate_batch(self):
" the 1970s",
" 7.1. 2",
" 13",
" evolution",
" stomach",
" spodumene",
" obama",
" northern new jersey",
" india",
" united stadium",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

Expand Down Expand Up @@ -905,13 +891,6 @@ def test_rag_sequence_generate_batch(self):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

Expand Down Expand Up @@ -957,13 +936,6 @@ def test_rag_sequence_generate_batch_from_context_input_ids(self):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

Expand Down

0 comments on commit bd8f6ca

Please sign in to comment.