not in a finally

Signed-off-by: Rob Howley <howley.robert@gmail.com>
robhowley · Oct 24, 2024 · ee2249a · ee2249a
1 parent 92dc4bd
commit ee2249a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/sdk/python/tests/integration/online_store/test_python_feature_server.py b/sdk/python/tests/integration/online_store/test_python_feature_server.py
@@ -136,10 +136,7 @@ def python_fs_client(environment, universal_data_sources, request):
     feast_objects.extend([driver(), customer(), location()])
     fs.apply(feast_objects)
     fs.materialize(environment.start_date, environment.end_date)
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(fs.initialize())
+    asyncio.get_event_loop().run_until_complete(fs.initialize())
     client = TestClient(get_app(fs))
-    try:
-        yield client
-    finally:
-        asyncio.get_event_loop().run_until_complete(fs.close())
+    yield client
+    asyncio.get_event_loop().run_until_complete(fs.close())