fbcode//security/genai/CybersecurityBenchmarks/benchmark/tests

Differential Revision: D52026102 fbshipit-source-id: b26b256309d5bc89c58371b3a42b01d9455b0ca1
meta-llama · Dec 11, 2023 · 5cf9ad6 · 5cf9ad6
1 parent cac8d15
commit 5cf9ad6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CybersecurityBenchmarks/benchmark/tests/test_llm.py b/CybersecurityBenchmarks/benchmark/tests/test_llm.py
@@ -13,7 +13,7 @@ class TestLLM(unittest.TestCase):
     def test_create(self) -> None:
         llama = llm.create("TOGETHER::togethercomputer/llama-2-7b::<my API key>")
         self.assertTrue(isinstance(llama, llm.TOGETHER))
-        self.assertEquals(llama.model, "togethercomputer/llama-2-7b")
-        self.assertEquals(llama.api_key, "<my API key>")
+        self.assertEqual(llama.model, "togethercomputer/llama-2-7b")
+        self.assertEqual(llama.api_key, "<my API key>")
 
         self.assertRaises(ValueError, lambda: llm.create("GPT::unknown::<my API key"))