graphnet.utilities.imports module

Common functions for icetray/data-based unit tests.

graphnet.utilities.imports.has_icecube_package()[source]

Check whether the icecube package is available.

Return type:

bool

graphnet.utilities.imports.has_torch_package()[source]

Check whether the torch package is available.

Return type:

bool

graphnet.utilities.imports.has_jammy_flows_package()[source]

Check if the jammy_flows package is available.

Return type:

bool

graphnet.utilities.imports.requires_icecube(test_function)[source]

Decorate test_function for use only if icecube module is present.

Return type:

Callable

Parameters:

test_function (Callable)