riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

make rag tests smaller (#10679)

Browse Source
This commit is contained in:
Patrick von Platen 2021-03-15 10:07:12 +03:00 committed by GitHub
parent 4c32f9f26e
commit bd8f6cafd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 0 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_modeling_rag.py
View File

@ -840,13 +840,6 @@ class RagModelIntegrationTests(unittest.TestCase):
"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
@ -885,13 +878,6 @@ class RagModelIntegrationTests(unittest.TestCase):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)
@ -942,13 +928,6 @@ class RagModelIntegrationTests(unittest.TestCase):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)
@ -986,13 +965,6 @@ class RagModelIntegrationTests(unittest.TestCase):
" the 1970s",
" 7.1. 2",
" 13",
" step by step",
" stomach",
" spodumene",
" obama",
" northern new jersey",
" india",
" united stadium",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)

28
tests/test_modeling_tf_rag.py
View File

@ -786,13 +786,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
"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
@ -861,13 +854,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
" the 1970s",
" 7.1. 2",
" 13",
" evolution",
" stomach",
" spodumene",
" obama",
" northern new jersey",
" india",
" united stadium",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)
@ -905,13 +891,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)
@ -957,13 +936,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
" 1980",
" 7.0",
" 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
]
self.assertListEqual(outputs, EXPECTED_OUTPUTS)