From 5cf9ad69743a63ad37c617e8fb0cca9ae2c9535a Mon Sep 17 00:00:00 2001
From: generatedunixname2443911735787003 <noreply+2443911735787003@fb.com>
Date: Mon, 11 Dec 2023 08:55:15 -0800
Subject: [PATCH] 
 fbcode//security/genai/CybersecurityBenchmarks/benchmark/tests

Differential Revision: D52026102

fbshipit-source-id: b26b256309d5bc89c58371b3a42b01d9455b0ca1
---
 CybersecurityBenchmarks/benchmark/tests/test_llm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CybersecurityBenchmarks/benchmark/tests/test_llm.py b/CybersecurityBenchmarks/benchmark/tests/test_llm.py
index 3ce709cf6..d0a6271db 100644
--- 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"))