启动垃圾收集时调试模式中的“ dataset. from generator” 出错

import gc
import tensorflow as tf
tf.data.experimental.enable_debug_mode()

def generator():
    yield 1
    yield 2
dataset = tf.data.Dataset.from_generator(
    generator, output_signature=tf.TensorSpec(shape=(), dtype=tf.int64)
)
iterator = iter(dataset)
print("first:", next(iterator).numpy())
gc.collect()
print("second:", next(iterator).numpy())  # ValueError: Could not find callback ...