import Sample in main __init__ (to make it discoverable for the doc)

blei-lab · Nov 4, 2024 · cf938d2 · cf938d2
1 parent 27e34be
commit cf938d2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/treeffuser/__init__.py b/src/treeffuser/__init__.py
@@ -1,4 +1,5 @@
 __version__ = "0.1.3"
+from treeffuser.samples import Samples
 from treeffuser.treeffuser import Treeffuser
 
-__all__ = ["Treeffuser"]
+__all__ = ["Treeffuser", "Samples"]