trw.utils.optional_import

Module Contents

Classes

_LazyRaise

Functions

optional_import(module_name: str, additional_error_message='')

Optional module import.

class trw.utils.optional_import._LazyRaise(msg)

__getattr__(self, name)

__call__(self, *args, **kwargs)

trw.utils.optional_import.optional_import(module_name: str, additional_error_message='')

Optional module import.

Raise an error only when a module is being used

Parameters

• module_name – the name of the module to import

• additional_error_message – add a custom error message

Returns

the module

Examples

>>> nn = optional_import('torch.nn')
>>> print(nn.ReLU)