remove reset, add updated notebook for quick check

theislab · Mar 9, 2024 · 7472eaa · 7472eaa
1 parent ee0b68f
commit 7472eaa
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 162 deletions.
diff --git a/cohort_tracking.ipynb b/cohort_tracking.ipynb
diff --git a/ehrapy/tools/cohort_tracking/_cohort_tracker.py b/ehrapy/tools/cohort_tracking/_cohort_tracker.py
@@ -105,16 +105,6 @@ def _get_cat_dicts(self, table_one, col):
     def _get_num_dicts(self, table_one, col):
         return table_one.cont_table["Overall"].loc[(col, "")]
 
-    def reset(self) -> None:
-        """Resets the `CohortTracker` object.
-
-        A full reset of the `CohortTracker` object.
-        """
-        self._tracked_steps = 0
-        self._track_t1 = []
-        self._tracked_text = []
-        self._tracked_operations = []
-
     @property
     def tracked_steps(self):
         """Number of tracked steps."""

diff --git a/tests/tools/cohort_tracking/test_cohort_tracking.py b/tests/tools/cohort_tracking/test_cohort_tracking.py
@@ -64,18 +64,6 @@ def test_CohortTracker_call(adata_mini):
     assert ct._tracked_text == ["Cohort 0\n (n=12)", "Cohort 1\n (n=12)"]
 
 
-def test_CohortTracker_reset(adata_mini):
-    ct = ep.tl.CohortTracker(adata_mini)
-
-    ct(adata_mini)
-    ct(adata_mini)
-
-    ct.reset()
-    assert ct.tracked_steps == 0
-    assert ct._tracked_text == []
-    assert ct._tracked_operations == []
-
-
 def test_CohortTracker_plot_cohort_change_test_sensitivity(adata_mini, check_same_image):
     ct = ep.tl.CohortTracker(adata_mini)