From da6ba7d591a0bf70b2018db30c4ff13388398d19 Mon Sep 17 00:00:00 2001 From: Grain Team Date: Mon, 1 Jun 2026 17:39:14 -0700 Subject: [PATCH] Fix slot memory leak when starting workers with Endoscope enabled. Old slot allocations are now reclaimed using weakref finalizers upon dataset iterator garbage collection. PiperOrigin-RevId: 925006068 --- .../_src/python/dataset/transformations/process_prefetch_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/grain/_src/python/dataset/transformations/process_prefetch_test.py b/grain/_src/python/dataset/transformations/process_prefetch_test.py index 610cb7eb2..fd97d3e80 100644 --- a/grain/_src/python/dataset/transformations/process_prefetch_test.py +++ b/grain/_src/python/dataset/transformations/process_prefetch_test.py @@ -14,6 +14,7 @@ from collections.abc import Sequence from concurrent import futures import dataclasses +import gc import logging as std_logging import os import sys