diff --git a/tests/test_main.py b/tests/test_main.py index a6a0b58..8f54f7c 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -265,7 +265,7 @@ def test_evaluate_groups_by_models(self, dai_instance: DAIndex) -> None: models=mock_model, feature_list=["feature_1", "feature_2"], reference_group="Male" ) - def test_get_plots(self, dai_instance: DAIndex) -> None: + def test_get_plot(self, dai_instance: DAIndex) -> None: """Test plot generation.""" # Mock the required data dai_instance.group_curves = { @@ -274,7 +274,7 @@ def test_get_plots(self, dai_instance: DAIndex) -> None: } dai_instance.group_ratios = {("Male", "Female"): {"Full": 0.1, "Decision": 0.05}} - fig = dai_instance.get_plots(reference_group="Male") + fig = dai_instance.get_plot(reference_group="Male") assert isinstance(fig, plt.Figure) def test_get_ratios(self, dai_instance: DAIndex) -> None: diff --git a/tutorials/meps_tutorial.ipynb b/tutorials/meps_tutorial.ipynb index c0223c7..e66761d 100644 --- a/tutorials/meps_tutorial.ipynb +++ b/tutorials/meps_tutorial.ipynb @@ -511,7 +511,7 @@ "metadata": {}, "outputs": [], "source": [ - "index.get_plots(reference_group=\"White\")" + "index.get_plot(reference_group=\"White\")" ] }, {