Skip to content

Commit 0db2282

Browse files
authored
Fix session cleanup (#1106)
1 parent 082c4e3 commit 0db2282

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/datachain/query/session.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -195,5 +195,11 @@ def _global_cleanup():
195195
Session.GLOBAL_SESSION_CTX.__exit__(None, None, None)
196196

197197
for obj in gc.get_objects(): # Get all tracked objects
198-
if isinstance(obj, Session): # Cleanup temp dataset for session variables.
199-
obj.__exit__(None, None, None)
198+
try:
199+
if isinstance(obj, Session):
200+
# Cleanup temp dataset for session variables.
201+
obj.__exit__(None, None, None)
202+
except ReferenceError:
203+
continue # Object has been finalized already
204+
except Exception as e: # noqa: BLE001
205+
logger.error(f"Exception while cleaning up session: {e}") # noqa: G004

0 commit comments

Comments
 (0)