From bd8f6cafd46b4b5b6d8f0c378c214775df031ab9 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 15 Mar 2021 10:07:12 +0300 Subject: [PATCH] make rag tests smaller (#10679) --- tests/test_modeling_rag.py | 28 ---------------------------- tests/test_modeling_tf_rag.py | 28 ---------------------------- 2 files changed, 56 deletions(-) diff --git a/tests/test_modeling_rag.py b/tests/test_modeling_rag.py index 5f69e1608de..6a31dcfa417 100644 --- a/tests/test_modeling_rag.py +++ b/tests/test_modeling_rag.py @@ -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) diff --git a/tests/test_modeling_tf_rag.py b/tests/test_modeling_tf_rag.py index 8dd1cb39d1b..679b25aa982 100644 --- a/tests/test_modeling_tf_rag.py +++ b/tests/test_modeling_tf_rag.py @@ -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)