rename treefusser to treeffuser

blei-lab · Feb 27, 2024 · 4891d24 · 4891d24
1 parent 9142b36
commit 4891d24
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/treeffuser/__init__.py b/src/treeffuser/__init__.py
@@ -1,4 +1,4 @@
 __version__ = "0.0.0"
-from treeffuser.treeffusser import LightGBMTreeffusser
+from treeffuser.treeffuser import LightGBMTreeffuser
 
-__all__ = ["LightGBMTreeffusser"]
+__all__ = ["LightGBMTreeffuser"]
diff --git a/src/treeffuser/treeffusser.py → src/treeffuser/treeffuser.py b/src/treeffuser/treeffusser.py → src/treeffuser/treeffuser.py
@@ -30,7 +30,7 @@ def _check_arguments(
     return
 
 
-class Treeffusser(BaseEstimator, abc.ABC):
+class Treeffuser(BaseEstimator, abc.ABC):
     """
     Abstract class for the Treeffuser model. Every particular
     score function has a slightly different implementation with
@@ -129,7 +129,7 @@ def sample(
         return y_transformed
 
 
-class LightGBMTreeffusser(Treeffusser):
+class LightGBMTreeffuser(Treeffuser):
     def __init__(
         self,
         # Diffusion model args

diff --git a/tests/test_treeffuser.py b/tests/test_treeffuser.py
@@ -1,7 +1,7 @@
 import numpy as np
 from scipy.stats import ks_2samp
 
-from treeffuser import LightGBMTreeffusser
+from treeffuser import LightGBMTreeffuser
 
 
 def test_treeffuser_bimodal_linear_regression():
@@ -32,7 +32,7 @@ def test_treeffuser_bimodal_linear_regression():
     X_test = X[n:]
     y_test = y[n:]
 
-    model = LightGBMTreeffusser(
+    model = LightGBMTreeffuser(
         verbose=1,
         n_repeats=100,
         n_estimators=10000,