"""Geometric datasets
This module provides a set of synthetic and real-world datasets
for evaluating neural networks on geometry tasks.
"""
from nugets.misc import import_submodules
[docs]
def get_dataset_register():
"""Get the register of all datasets
This function imports all the datasets in the current module.
It is gated behind a function to avoid doing that at import time.
"""
from .register import register
#### import all the datasets to populate the register
import_submodules(__name__)
return register