correctly reference validators

Author: zsimjee

tests/integration_tests/validators/test_on_topic.py

@@ -16,7 +16,7 @@ def test_validate_valid_topic_cpu_disable_llm(self):
         )
         text = "This is an article about sports."
         expected_result = PassResult()
-        actual_result = validator.validate(text)
+        actual_result = validator.validate(text, metadata={})
         self.assertEqual(actual_result, expected_result)
 
     def test_validate_invalid_topic_cpu_disable_llm(self):
@@ -28,7 +28,7 @@ def test_validate_invalid_topic_cpu_disable_llm(self):
         )
         text = "This is an article about music."
         expected_result = FailResult(error_message="Most relevant topic is other.")
-        actual_result = validator.validate(text)
+        actual_result = validator.validate(text, metadata={})
         self.assertEqual(actual_result, expected_result)
 
 
@@ -72,7 +72,7 @@ def test_validate_valid_topic_cpu_enable_llm(self):
         )
         text = "This is a historical analysis of political events."
         expected_result = PassResult()
-        actual_result = validator.validate(text)
+        actual_result = validator.validate(text, metadata={})
         self.assertEqual(actual_result, expected_result)
 
     def test_validate_invalid_topic_cpu_enable_llm(self):
@@ -84,7 +84,7 @@ def test_validate_invalid_topic_cpu_enable_llm(self):
         )
         text = "This is an article about cooking recipes."
         expected_result = FailResult(error_message="Most relevant topic is other.")
-        actual_result = validator.validate(text)
+        actual_result = validator.validate(text, metadata={})
         self.assertEqual(actual_result, expected_result)
 
 
@@ -101,7 +101,7 @@ def test_validate_valid_topic_disable_model(self):
         )
         text = "This is a movie review for the new sports action film."
         expected_result = PassResult()
-        actual_result = validator.validate(text)
+        actual_result = validator.validate(text, metadata={})
         self.assertEqual(actual_result, expected_result)
 
     def test_validate_invalid_topic_disable_model(self):
@@ -112,5 +112,5 @@ def test_validate_invalid_topic_disable_model(self):
         )
         text = "This is a research paper on medical advancements."
         expected_result = FailResult(error_message="Most relevant topic is other.")
-        actual_result = validator.validate(text)
+        actual_result = validator.validate(text, metadata={})
         self.assertEqual(actual_result, expected_result)

tests/unit_tests/validators/test_on_topic.py

@@ -113,7 +113,7 @@ def test_get_topic_zero_shot(self):
     def test_validate_valid_topic(self):
         text = "This is an article about sports."
         validator = OnTopic(valid_topics=self.valid_topics)
-        validation_result = validator.validate(text)
+        validation_result = validator.validate(text, metadata={})
         self.assertEqual(validation_result, PassResult())
 
     def test_validate_invalid_topic(self):
@@ -122,7 +122,7 @@ def test_validate_invalid_topic(self):
             mock_llm.return_value = '{"topic": "other"}'
 
             text = "This is an article about music."
-            validation_result = validator.validate(text)
+            validation_result = validator.validate(text, metadata={})
 
             self.assertEqual(
                 validation_result,
@@ -132,9 +132,9 @@ def test_validate_invalid_topic(self):
     def test_validate_no_valid_topics(self):
         with self.assertRaises(ValueError):
             validator = OnTopic(valid_topics=[])
-            validator.validate("This is a test text.")
+            validator.validate("This is a test text.", metadata={})
 
     def test_validate_overlapping_topics(self):
         with self.assertRaises(ValueError):
             validator = OnTopic(valid_topics=["sports"], invalid_topics=["sports"])
-            validator.validate("This is a test text.")
+            validator.validate("This is a test text.", metadata={})