# Copyright Contributors to the Pyro project.
# SPDX-License-Identifier: Apache-2.0
import pyroapi
from pyro.primitives import ( # noqa: F401
clear_param_store, deterministic, enable_validation, factor, get_param_store,
module, param, random_module, sample, set_rng_seed, subsample,
)
from pyro.contrib.funsor.handlers.primitives import to_data, to_funsor # noqa: F401
from pyro.contrib.funsor.handlers import condition, do, markov # noqa: F401
from pyro.contrib.funsor.handlers import plate as _plate
[docs]def plate(*args, **kwargs):
return _plate(None, *args, **kwargs)
pyroapi.register_backend('contrib.funsor', {
'distributions': 'pyro.distributions',
'handlers': 'pyro.contrib.funsor.handlers',
'infer': 'pyro.contrib.funsor.infer',
'ops': 'torch',
'optim': 'pyro.optim',
'pyro': 'pyro.contrib.funsor',
})