Files
cdtools/CDTools/__init__.py
T

13 lines
346 B
Python

# This is needed to allow us to use torch.tensor in the module without it
# constantly complaining.
import warnings
warnings.filterwarnings("ignore",
message='To copy construct from a tensor, ')
__all__ = ['tools', 'datasets', 'models']
from CDTools import tools
from CDTools import datasets
from CDTools import models