lightning_uq_box.datamodules
============================

.. module:: lightning_uq_box.datamodules

Toy Datamodules
---------------

.. autoclass:: ToyHeteroscedasticDatamodule


Image Regression Datamodules
----------------------------

.. autoclass:: ToyImageRegressionDatamodule