Method add_import

tools/Polygraphy/polygraphy/tools/script.py:270–294

Adds imports to this script. For example: :: script.add_import("numpy") # Equivalent to: import numpy script.add_import("numpy", imp_as="np") # Equivalent to: import numpy as np script.add_import("TrtRunner", frm="polygraphy.backend.trt"

(self, imports, frm=None, imp_as=None)

        self.always_create_runners = always_create_runners

    def add_import(self, imports, frm=None, imp_as=None):
        """
        Adds imports to this script.

        For example:
        ::

            script.add_import("numpy") # Equivalent to: import numpy
            script.add_import("numpy", imp_as="np") # Equivalent to: import numpy as np
            script.add_import("TrtRunner", frm="polygraphy.backend.trt") # Equivalent to: from polygraphy.backend.trt import TrtRunner

        Args:
            imports (Union[str, List[str]]): Object/submodule or list of objects/submodules to import.
            frm (str): Module from which to import.
            imp_as (str): Name to import as. When this is specified, ``imports`` must be a string and not a list.
        """
        if isinstance(imports, str):
            imports = {imports}

        if imp_as and len(imports) > 1:
            G_LOGGER.internal_error("When `imp_as` is specified, `imports` must be a string and not a list")

        if frm not in self.imports:
            self.imports[frm] = set()
        self.imports[frm].update({(imp, imp_as) for imp in imports})

    def set_data_loader(self, data_loader_str):
        """