rename scaler.py

blei-lab · Nov 4, 2024 · 27e34be · 27e34be
1 parent be5c61d
commit 27e34be
Show file tree

Hide file tree

Showing 7 changed files with 6 additions and 6 deletions.
diff --git a/src/treeffuser/_base_tabular_diffusion.py b/src/treeffuser/_base_tabular_diffusion.py
@@ -13,9 +13,9 @@
 from sklearn.neighbors import KernelDensity
 from tqdm import tqdm
 
+from treeffuser._scaler import ScalerMixedTypes
 from treeffuser._score_models import ScoreModel
 from treeffuser._warnings import ConvergenceWarning
-from treeffuser.scaler import ScalerMixedTypes
 from treeffuser.sde import DiffusionSDE
 from treeffuser.sde import sdeint
 

diff --git a/src/treeffuser/scaler.py → src/treeffuser/_scaler.py b/src/treeffuser/scaler.py → src/treeffuser/_scaler.py
diff --git a/testbed/src/testbed/models/lightning_uq_models/_card.py b/testbed/src/testbed/models/lightning_uq_models/_card.py
@@ -29,7 +29,7 @@
 from testbed.models.base_model import ProbabilisticModel
 from testbed.models.lightning_uq_models._data_module import GenericDataModule
 from testbed.models.lightning_uq_models._utils import _to_tensor
-from treeffuser.scaler import ScalerMixedTypes
+from treeffuser._scaler import ScalerMixedTypes
 
 
 class Card(ProbabilisticModel, MultiOutputMixin):

diff --git a/testbed/src/testbed/models/lightning_uq_models/_deep_ensemble.py b/testbed/src/testbed/models/lightning_uq_models/_deep_ensemble.py
@@ -20,7 +20,7 @@
 
 from testbed.models.base_model import ProbabilisticModel
 from testbed.models.lightning_uq_models._data_module import GenericDataModule
-from treeffuser.scaler import ScalerMixedTypes
+from treeffuser._scaler import ScalerMixedTypes
 
 
 class MLP(nn.Module):

diff --git a/testbed/src/testbed/models/lightning_uq_models/_mc_dropout.py b/testbed/src/testbed/models/lightning_uq_models/_mc_dropout.py
@@ -20,7 +20,7 @@
 from testbed.models.base_model import ProbabilisticModel
 from testbed.models.lightning_uq_models._data_module import GenericDataModule
 from testbed.models.lightning_uq_models._utils import _to_tensor
-from treeffuser.scaler import ScalerMixedTypes
+from treeffuser._scaler import ScalerMixedTypes
 
 
 class MCDropout(ProbabilisticModel, MultiOutputMixin):

diff --git a/testbed/src/testbed/models/lightning_uq_models/_quantile_regression.py b/testbed/src/testbed/models/lightning_uq_models/_quantile_regression.py
@@ -17,7 +17,7 @@
 from testbed.models.base_model import ProbabilisticModel
 from testbed.models.lightning_uq_models._data_module import GenericDataModule
 from testbed.models.lightning_uq_models._utils import _to_tensor
-from treeffuser.scaler import ScalerMixedTypes
+from treeffuser._scaler import ScalerMixedTypes
 
 
 class QuantileRegression(ProbabilisticModel):

diff --git a/testbed/src/testbed/models/ngboost/_ngboost.py b/testbed/src/testbed/models/ngboost/_ngboost.py
@@ -11,7 +11,7 @@
 
 from testbed.models.base_model import ProbabilisticModel
 from testbed.models.ngboost._gaussian_mixtures import build_gaussian_mixture_model
-from treeffuser.scaler import ScalerMixedTypes
+from treeffuser._scaler import ScalerMixedTypes
 
 MAX_MINIBATCH_SIZE = 50_000
 MAX_VALIDATION_SIZE = 20_000