message: [tf.data] Checking for py_func using node op instead of node name.
diff: @@ -2799,7 +2799,7 @@ class StructuredFunctionWrapper(object):
  # triggers use-after-free. Figure out why and stop excluding functions
  # with PyFunc nodes from garbage collection.
  for node in self._function.function_def.node_def:
  - if "PyFunc" in node.name:
  + if node.op in ("PyFunc", "PyFuncStateless", "EagerPyFunc"):
  self._function._garbage_collector.release()
  if add_to_graph:
  