hotfix: memoized volume object not reusable

Backported-from: main (24.09) Backported-to: 23.09
lablup · Apr 29, 2024 · 55a34cf · 55a34cf
1 parent 55dfd69
commit 55a34cf
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/ai/backend/storage/context.py b/src/ai/backend/storage/context.py
@@ -200,7 +200,4 @@ async def get_volume(self, name: str) -> AsyncIterator[AbstractVolume]:
             await volume_obj.init()
             self.volumes[name] = volume_obj
 
-            try:
-                yield volume_obj
-            finally:
-                await volume_obj.shutdown()
+            yield volume_obj