phast.utils#

Functions#

ensure_pyg_ok(func)

Decorator to ensure that torch_geometric is installed when

Attributes#

PYG_OK

phast.utils.PYG_OK = True[source]#
phast.utils.ensure_pyg_ok(func)[source]#

Decorator to ensure that torch_geometric is installed when using a function that requires it.

Parameters

func (callable) – Function to decorate.