Skip to content

GitLab

Unverified Commit bd8f6caf authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

make rag tests smaller (#10679)

parent 4c32f9f2
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 56 deletions
tests/test_modeling_rag.py
View file @ bd8f6caf
...@@ -840,13 +840,6 @@ class RagModelIntegrationTests(unittest.TestCase): ...@@ -840,13 +840,6 @@ class RagModelIntegrationTests(unittest.TestCase):
"when is the last time the philadelphia won the superbowl", "when is the last time the philadelphia won the superbowl",
"what is the most current adobe flash player version", "what is the most current adobe flash player version",
"how many episodes are there in dragon ball z", "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 @slow
...@@ -885,13 +878,6 @@ class RagModelIntegrationTests(unittest.TestCase): ...@@ -885,13 +878,6 @@ class RagModelIntegrationTests(unittest.TestCase):
" 1980", " 1980",
" 7.0", " 7.0",
" 8", " 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
] ]
self.assertListEqual(outputs, EXPECTED_OUTPUTS) self.assertListEqual(outputs, EXPECTED_OUTPUTS)
...@@ -942,13 +928,6 @@ class RagModelIntegrationTests(unittest.TestCase): ...@@ -942,13 +928,6 @@ class RagModelIntegrationTests(unittest.TestCase):
" 1980", " 1980",
" 7.0", " 7.0",
" 8", " 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
] ]
self.assertListEqual(outputs, EXPECTED_OUTPUTS) self.assertListEqual(outputs, EXPECTED_OUTPUTS)
...@@ -986,13 +965,6 @@ class RagModelIntegrationTests(unittest.TestCase): ...@@ -986,13 +965,6 @@ class RagModelIntegrationTests(unittest.TestCase):
" the 1970s", " the 1970s",
" 7.1. 2", " 7.1. 2",
" 13", " 13",
" step by step",
" stomach",
" spodumene",
" obama",
" northern new jersey",
" india",
" united stadium",
] ]
self.assertListEqual(outputs, EXPECTED_OUTPUTS) self.assertListEqual(outputs, EXPECTED_OUTPUTS)
......
tests/test_modeling_tf_rag.py
View file @ bd8f6caf
...@@ -786,13 +786,6 @@ class TFRagModelIntegrationTests(unittest.TestCase): ...@@ -786,13 +786,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
"when is the last time the philadelphia won the superbowl", "when is the last time the philadelphia won the superbowl",
"what is the most current adobe flash player version", "what is the most current adobe flash player version",
"how many episodes are there in dragon ball z", "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 @slow
...@@ -861,13 +854,6 @@ class TFRagModelIntegrationTests(unittest.TestCase): ...@@ -861,13 +854,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
" the 1970s", " the 1970s",
" 7.1. 2", " 7.1. 2",
" 13", " 13",
" evolution",
" stomach",
" spodumene",
" obama",
" northern new jersey",
" india",
" united stadium",
] ]
self.assertListEqual(outputs, EXPECTED_OUTPUTS) self.assertListEqual(outputs, EXPECTED_OUTPUTS)
...@@ -905,13 +891,6 @@ class TFRagModelIntegrationTests(unittest.TestCase): ...@@ -905,13 +891,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
" 1980", " 1980",
" 7.0", " 7.0",
" 8", " 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
] ]
self.assertListEqual(outputs, EXPECTED_OUTPUTS) self.assertListEqual(outputs, EXPECTED_OUTPUTS)
...@@ -957,13 +936,6 @@ class TFRagModelIntegrationTests(unittest.TestCase): ...@@ -957,13 +936,6 @@ class TFRagModelIntegrationTests(unittest.TestCase):
" 1980", " 1980",
" 7.0", " 7.0",
" 8", " 8",
" reticular formation",
" walls of the abdomen",
" spodumene",
" obama",
" new orleans",
" japan",
" old trafford",
] ]
self.assertListEqual(outputs, EXPECTED_OUTPUTS) self.assertListEqual(outputs, EXPECTED_OUTPUTS)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment