From d6e520489d1bbdcd57613a46c02fb2cf8afff3be Mon Sep 17 00:00:00 2001
From: Anji Liu <anjiliu219@gmail.com>
Date: Wed, 6 Mar 2024 17:54:39 +0800
Subject: [PATCH] make alias `pyjuice.distributions`

---
 src/pyjuice/__init__.py               | 1 +
 src/pyjuice/distributions/__init__.py | 3 +++
 2 files changed, 4 insertions(+)
 create mode 100644 src/pyjuice/distributions/__init__.py

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