Minimize re-exports from __init__ files (#44)

This allows importing parts of the package without having to
import practically everything (since importing a package will
import its parents' __init__s, etc).
This commit is contained in:
Aarni Koskela
2023-07-25 17:24:09 +03:00
committed by GitHub
parent ef520df1db
commit 57862fb4c7
5 changed files with 7 additions and 14 deletions

View File

@@ -1,7 +0,0 @@
from .denoiser import Denoiser
from .discretizer import Discretization
from .loss import StandardDiffusionLoss
from .model import Model, Encoder, Decoder
from .openaimodel import UNetModel
from .sampling import BaseDiffusionSampler
from .wrappers import OpenAIWrapper