Expose API functions from top-level (#7)

Also update cookiecutter package in cthoyt/cookiecutter-snekpack@fa032ff to have this be a part by default
mberr · May 10, 2022 · 85d4884 · 85d4884
1 parent 5e1a0cb
commit 85d4884
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/torch_ppr/__init__.py b/src/torch_ppr/__init__.py
@@ -1,3 +1,10 @@
 # -*- coding: utf-8 -*-
 
 """(Personalized) Page-Rank computation using PyTorch."""
+
+from .api import page_rank, personalized_page_rank
+
+__all__ = [
+    "page_rank",
+    "personalized_page_rank",
+]