diff --git a/tests/cores/mega/test_dynamic_batching.py b/tests/cores/mega/test_dynamic_batching.py index dafc6f911..bcb185b8f 100644 --- a/tests/cores/mega/test_dynamic_batching.py +++ b/tests/cores/mega/test_dynamic_batching.py @@ -3,7 +3,6 @@ import asyncio import multiprocessing -import os import unittest from enum import Enum @@ -77,7 +76,6 @@ def tearDown(self): self.process1.terminate() async def test_dynamic_batching(self): - os.environ["LOGFLAG"] = True url1 = "http://localhost:8080/v1/add1" url2 = "http://localhost:8080/v1/add2" diff --git a/tests/cores/mega/test_hybrid_service_orchestrator.py b/tests/cores/mega/test_hybrid_service_orchestrator.py index cd4c198fb..89522eac3 100644 --- a/tests/cores/mega/test_hybrid_service_orchestrator.py +++ b/tests/cores/mega/test_hybrid_service_orchestrator.py @@ -34,7 +34,7 @@ def test_add_remote_service(self): self.service_builder.add(opea_microservices["s1"]).add(s2) self.service_builder.flow_to(self.s1, s2) self.assertEqual(s2.endpoint_path, "http://fakehost:8008/v1/add") - self.assertRaises(Exception, s2._validate_env(), "N/A") + self.assertRaises(Exception, s2._validate_env, "N/A") if __name__ == "__main__":