only clean-up azurite containers locally

microsoft · Apr 2, 2024 · 56e582b · 56e582b
1 parent 32537df
commit 56e582b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/graphrag/tests/smoke/test_fixtures.py b/python/graphrag/tests/smoke/test_fixtures.py
@@ -112,7 +112,11 @@ async def prepare_azurite_data(input_path: str, azure: dict) -> Callable[[], Non
         )
         await input_storage.set(file_path, text, encoding="utf-8")
 
-    return lambda: input_storage.delete_container()
+    def dispose():
+        if BLOB_CONNECTION_STRING == WELL_KNOWN_AZURITE_CONNECTION_STRING:
+            input_storage.delete_container()
+
+    return dispose
 
 
 class TestIndexer: