From 27e34be15c64c18cca2fe79902df89ac4b43da6d Mon Sep 17 00:00:00 2001
From: Achille Nazaret <aon2108@columbia.edu>
Date: Mon, 4 Nov 2024 15:34:05 -0500
Subject: [PATCH] rename scaler.py

---
 src/treeffuser/_base_tabular_diffusion.py                       | 2 +-
 src/treeffuser/{scaler.py => _scaler.py}                        | 0
 testbed/src/testbed/models/lightning_uq_models/_card.py         | 2 +-
 .../src/testbed/models/lightning_uq_models/_deep_ensemble.py    | 2 +-
 testbed/src/testbed/models/lightning_uq_models/_mc_dropout.py   | 2 +-
 .../testbed/models/lightning_uq_models/_quantile_regression.py  | 2 +-
 testbed/src/testbed/models/ngboost/_ngboost.py                  | 2 +-
 7 files changed, 6 insertions(+), 6 deletions(-)
 rename src/treeffuser/{scaler.py => _scaler.py} (100%)

diff --git a/src/treeffuser/_base_tabular_diffusion.py b/src/treeffuser/_base_tabular_diffusion.py
index bda05032..84fa48d8 100644
--- 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 b/src/treeffuser/_scaler.py
similarity index 100%
rename from src/treeffuser/scaler.py
rename to 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
index 8a4c4f41..b2393856 100644
--- 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
index 2391be0c..1c139d6e 100644
--- 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
index 194b0f5f..bb009934 100644
--- 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
index ff7589f8..358f2946 100644
--- 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
index 3be80548..3026acfd 100644
--- 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