fix test case

georgia-tech-db · Oct 16, 2023 · a4276ae · a4276ae
1 parent c84e0d7
commit a4276ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/unit_tests/test_dalle.py b/test/unit_tests/test_dalle.py
@@ -41,7 +41,7 @@ def setUp(self) -> None:
     def tearDown(self) -> None:
         execute_query_fetch_all(self.evadb, "DROP TABLE IF EXISTS ImageGen;")
 
-    @patch.dict("os.environ", {"OPENAI_KEY": "mocked_openai_key"})
+    @patch.dict("os.environ", {"OPENAI_API_KEY": "mocked_openai_key"})
     @patch("requests.get")
     @patch("openai.Image.create", return_value={"data": [{"url": "mocked_url"}]})
     def test_dalle_image_generation(self, mock_openai_create, mock_requests_get):