fixing name change bug

owkin · Nov 28, 2023 · 8b2a744 · 8b2a744
1 parent 96a0a4c
commit 8b2a744
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 8 deletions.
diff --git a/docs/source/api/iptw.rst b/docs/source/api/iptw.rst
@@ -1,4 +1,4 @@
-fedeca.fedeca
+fedeca.fedeca_core
 =========================
 
 .. autoclass:: fedeca.FedECA
diff --git a/experiments/config/model/fl_iptw.yaml b/experiments/config/model/fl_iptw.yaml
@@ -1,6 +1,6 @@
 defaults:
   - common
 
-_target_: fedeca.fedeca.FedECA
+_target_: fedeca.fedeca_core.FedECA
 num_rounds_list: [20, 20]
 dtype: "float64"
diff --git a/fedeca/scripts/dp_logreg.py b/fedeca/scripts/dp_logreg.py
@@ -13,7 +13,7 @@
 from torch.optim import SGD
 
 from fedeca.algorithms.torch_dp_fed_avg_algo import TorchDPFedAvgAlgo
-from fedeca.fedeca import LogisticRegressionTorch
+from fedeca.fedeca_core import LogisticRegressionTorch
 from fedeca.utils import (
     Experiment,
     make_accuracy_function,

diff --git a/fedeca/tests/test_dp_end2end.py b/fedeca/tests/test_dp_end2end.py
@@ -10,7 +10,7 @@
 from torch.optim import SGD
 
 from fedeca.algorithms.torch_dp_fed_avg_algo import TorchDPFedAvgAlgo
-from fedeca.fedeca import LogisticRegressionTorch
+from fedeca.fedeca_core import LogisticRegressionTorch
 from fedeca.tests.common import TestTempDir
 from fedeca.utils import (
     Experiment,

diff --git a/fedeca/tests/test_fliptw_end2end.py b/fedeca/tests/test_fliptw_end2end.py
@@ -1,7 +1,7 @@
 """Tests for webdisco."""
 from pandas.testing import assert_frame_equal
 
-from fedeca.fedeca import FedECA
+from fedeca.fedeca_core import FedECA
 from fedeca.tests.common import TestTempDir
 from fedeca.utils.data_utils import generate_survival_data
 

diff --git a/fedeca/tests/test_robust_iptw.py b/fedeca/tests/test_robust_iptw.py
@@ -3,7 +3,7 @@
 import pytest
 
 from fedeca.competitors import PooledIPTW
-from fedeca.fedeca import FedECA
+from fedeca.fedeca_core import FedECA
 from fedeca.tests.common import TestTempDir
 from fedeca.utils.survival_utils import CoxData
 

diff --git a/fedeca/utils/experiment_utils.py b/fedeca/utils/experiment_utils.py
@@ -12,7 +12,7 @@
 import pandas as pd
 from lifelines.exceptions import ConvergenceError
 
-from fedeca.fedeca import FedECA
+from fedeca.fedeca_core import FedECA
 from fedeca.utils.survival_utils import BaseSurvivalEstimator, CoxData
 from fedeca.utils.typing import _SeedType
 
@@ -127,7 +127,7 @@ def single_experiment(
         The seed for reproducibility. Defaults to None.
     fit_fedeca: dict[str, Any], optional, by default None
         Dictionary of kwargs for the fit function of
-        :class:`fedeca.fedeca.FedECA`.
+        :class:`fedeca.fedeca_core.FedECA`.
     return_propensities: bool
         return propensity scores in the results dataframe, by default True
     return_weights: bool