diff --git a/src/pyjuice/__init__.py b/src/pyjuice/__init__.py index 364bdd49..8ad855f8 100644 --- a/src/pyjuice/__init__.py +++ b/src/pyjuice/__init__.py @@ -1,5 +1,6 @@ import pyjuice.graph import pyjuice.nodes +import pyjuice.distributions import pyjuice.layer import pyjuice.structures import pyjuice.optim diff --git a/src/pyjuice/distributions/__init__.py b/src/pyjuice/distributions/__init__.py new file mode 100644 index 00000000..fef80366 --- /dev/null +++ b/src/pyjuice/distributions/__init__.py @@ -0,0 +1,3 @@ +# `pyjuice.distributions` is an alias of `pyjuice.nodes.distributions` + +from pyjuice.nodes.distributions import * \ No newline at end of file