diff --git a/pplm_affinity/__init__.py b/pplm_affinity/__init__.py
new file mode 100644
index 0000000..c66c577
--- /dev/null
+++ b/pplm_affinity/__init__.py
@@ -0,0 +1 @@
+from .model import PPLM_Affinity