Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Tests] RAG #10679

Merged
merged 1 commit into from
Mar 15, 2021
Merged

[Tests] RAG #10679

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
28 changes: 0 additions & 28 deletions tests/test_modeling_rag.py
View file
Open in desktop
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
Open in desktop
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 @@ -907,13 +893,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 @@ -961,13 +940,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