Global Configuration

Module: polygraphy.config

INTERNAL_CORRECTNESS_CHECKS = False

Whether internal correctness checks are enabled. This can be configured by setting the ‘POLYGRAPHY_INTERNAL_CORRECTNESS_CHECKS’ environment variable.

Type

bool

AUTOINSTALL_DEPS = False

Whether Polygraphy will automatically install required Python packages at runtime. This can be configured by setting the ‘POLYGRAPHY_AUTOINSTALL_DEPS’ environment variable.

Type

bool

INSTALL_CMD = ['/usr/bin/python3', '-m', 'pip', 'install']

The command to use to automatically install dependencies. Only relevant when AUTOINSTALL_DEPS is enabled. Defaults to ["python", "-m", "pip", "install"]. This can be configured by setting the ‘POLYGRAPHY_INSTALL_CMD’ environment variable to a string containing the command; for example: python3 -m pip install.

Type

List[str]

ARRAY_SWAP_THRESHOLD_MB = -1

The threshold, in megabytes, above which Polygraphy will evict a NumPy array from memory and swap it to disk. A negative value disables swapping and a value of 0 causes all arrays to be saved to disk. Disabled by default. This can be configured by setting the ‘POLYGRAPHY_ARRAY_SWAP_THRESHOLD_MB’ environment variable.

Type

int