Change plots submodule name to interpret (#705)

Co-authored-by: Osvaldo A Martin <[email protected]> Co-authored-by: Tomas Capretto <[email protected]>
bambinos · Aug 17, 2023 · 00f415c · 00f415c
1 parent 5dde496
commit 00f415c
Show file tree

Hide file tree

Showing 13 changed files with 1,974 additions and 2,881 deletions.
diff --git a/bambi/__init__.py b/bambi/__init__.py
@@ -9,6 +9,7 @@
 from .models import Model
 from .priors import Prior
 from .version import __version__
+from . import interpret
 
 
 __all__ = [

diff --git a/bambi/interpret/__init__.py b/bambi/interpret/__init__.py
@@ -0,0 +1,11 @@
+from bambi.interpret.effects import comparisons, predictions, slopes
+from bambi.interpret.plotting import plot_comparisons, plot_predictions, plot_slopes
+
+__all__ = [
+    "comparisons",
+    "slopes",
+    "predictions",
+    "plot_comparisons",
+    "plot_predictions",
+    "plot_slopes",
+]
diff --git a/bambi/plots/create_data.py → bambi/interpret/create_data.py b/bambi/plots/create_data.py → bambi/interpret/create_data.py
@@ -4,7 +4,7 @@
 import pandas as pd
 
 from bambi.models import Model
-from bambi.plots.utils import (
+from bambi.interpret.utils import (
     ConditionalInfo,
     enforce_dtypes,
     get_covariates,

diff --git a/bambi/plots/effects.py → bambi/interpret/effects.py b/bambi/plots/effects.py → bambi/interpret/effects.py
@@ -10,8 +10,8 @@
 import xarray as xr
 
 from bambi.models import Model
-from bambi.plots.create_data import create_differences_data, create_predictions_data
-from bambi.plots.utils import (
+from bambi.interpret.create_data import create_differences_data, create_predictions_data
+from bambi.interpret.utils import (
     average_over,
     ConditionalInfo,
     identity,

diff --git a/bambi/plots/plot_types.py → bambi/interpret/plot_types.py b/bambi/plots/plot_types.py → bambi/interpret/plot_types.py
@@ -3,7 +3,7 @@
 import numpy as np
 import pandas as pd
 
-from bambi.plots.utils import Covariates, get_unique_levels, get_group_offset, identity
+from bambi.interpret.utils import Covariates, get_unique_levels, get_group_offset, identity
 
 
 def plot_numeric(

diff --git a/bambi/plots/plotting.py → bambi/interpret/plotting.py b/bambi/plots/plotting.py → bambi/interpret/plotting.py
@@ -8,9 +8,9 @@
 from pandas.api.types import is_categorical_dtype, is_numeric_dtype, is_string_dtype
 
 from bambi.models import Model
-from bambi.plots.effects import comparisons, slopes, predictions
-from bambi.plots.plot_types import plot_categoric, plot_numeric
-from bambi.plots.utils import get_covariates, ConditionalInfo
+from bambi.interpret.effects import comparisons, slopes, predictions
+from bambi.interpret.plot_types import plot_categoric, plot_numeric
+from bambi.interpret.utils import get_covariates, ConditionalInfo
 from bambi.utils import get_aliased_name, listify
 
 

diff --git a/bambi/plots/utils.py → bambi/interpret/utils.py b/bambi/plots/utils.py → bambi/interpret/utils.py
diff --git a/bambi/plots/__init__.py b/bambi/plots/__init__.py
@@ -1,5 +1,5 @@
-from bambi.plots.effects import comparisons, predictions, slopes
-from bambi.plots.plotting import plot_comparisons, plot_predictions, plot_slopes
+from bambi.interpret.effects import comparisons, predictions, slopes
+from bambi.interpret.plotting import plot_comparisons, plot_predictions, plot_slopes
 
 
 __all__ = [

diff --git a/docs/notebooks/plot_comparisons.ipynb b/docs/notebooks/plot_comparisons.ipynb
diff --git a/docs/notebooks/plot_predictions.ipynb b/docs/notebooks/plot_predictions.ipynb