def _setup_input_callbacks(self):
from nvidia.dali.external_source import _is_external_source_with_callback
groups = set()
for op in self._ops:
if _is_external_source_with_callback(op):
group = op._group
groups.add(group)
self._input_callbacks = list(groups)
def _get_name_es_instance_map(self):
"""Return mappings between name of External Source and the op.
Returns
-------
mapping for placeholders nodes, mapping for nodes with Python source
Two mappings are returned, separating the placeholder nodes without a `source`
and nodes that got a `source` parameter.
"""
name_es = {}
name_es_with_callback = {}
for op in self._pipeline_instance._ops:
if _is_external_source_with_callback(op):
# use the internal op name (generated automatically in most cases)
name_es_with_callback[op.name] = op
elif _is_external_source(op):
self._assert_correct_external_sources(op)
# use the user provided name
name_es[op._op._name] = op
return name_es, name_es_with_callback
