Function import_numpy

dali/python/nvidia/dali/_utils/external_source_impl.py:25–35

Import numpy lazily, need to define global `np = None` variable

()

def import_numpy():
    """Import numpy lazily, need to define global `np = None` variable"""
    global np
    if np is None:
        try:
            import numpy as np
        except ImportError:
            raise RuntimeError(
                "Could not import numpy. Please make sure you have numpy "
                "installed before you use parallel mode."
            )


class SourceKind(Enum):