add ploomber-extension (#342)

* adds ploomber-extension as a dependency

* linting

* fix tests

* update baseline images

examples/conftest.py

@@ -1,6 +1,7 @@
 """
 Test configuration file, ignore
 """
+
 import matplotlib.pyplot as plt
 
 # mock plt.show() so plots do not appear when running the tests

examples/data_grid.py

@@ -2,6 +2,7 @@
 %load_ext autoreload
 %autoreload 2
 """
+
 import matplotlib.pyplot as plt
 from sklearn_evaluation.plot.matplotlib.bar import bar_groups
 from sklearn_evaluation.plot.matplotlib.data_grid import DataGrid

examples/grouped_bar.py

@@ -1,5 +1,6 @@
 """
 """
+
 import matplotlib.pyplot as plt
 from sklearn_evaluation.plot.matplotlib.bar import bar_groups
 

examples/heatmap.py

@@ -1,5 +1,6 @@
 """
 """
+
 import matplotlib.pyplot as plt
 from sklearn_evaluation.plot.matplotlib import heatmap
 

override_failed_result_images.py

@@ -3,6 +3,7 @@
 this happens sometimes when we make changes, we have to check manually
 if differences are acceptable, then run this script
 """
+
 from pathlib import Path
 from glob import glob
 import shutil

setup.py

@@ -32,6 +32,10 @@
     "altair",
     # for pandas.read_html (optional dependency for NotebookCollection)
     "lxml",
+    # tmp fix, 3.8 breaks the tests:
+    # ERROR tests/test_grid_search.py - ImportError: cannot import name 'cleanup'
+    # from 'matplotlib.testing.decorators'
+    "matplotlib<3.8",
 ]
 
 
@@ -54,6 +58,7 @@
     include_package_data=True,
     install_requires=[
         "ploomber-core>=0.2.6",
+        "ploomber-extension",
         # compute metrics
         "scikit-learn",
         # plotting

src/conftest.py

@@ -1,6 +1,7 @@
 """
 Test configuration for doctests
 """
+
 import os
 from pathlib import Path
 

src/sklearn_evaluation/plot/_example.py

@@ -6,6 +6,7 @@
 
 .. plot:: documentation https://matplotlib.org/3.1.3/devel/plot_directive.html
 """
+
 from collections import Counter
 
 

src/sklearn_evaluation/plot/calibration.py

@@ -25,6 +25,7 @@
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 """
+
 import numpy as np
 import matplotlib.pyplot as plt
 from sklearn.calibration import calibration_curve as sk_calibration_curve
@@ -111,7 +112,6 @@ def _generate_colors(cmap, n_color):
 
 
 class CalibrationCurve(AbstractPlot):
-
     """
     Parameters
     ----------

src/sklearn_evaluation/plot/classification.py

@@ -1,6 +1,7 @@
 """
 Plotting functions for classifier models
 """
+
 import json
 from pathlib import Path
 

src/sklearn_evaluation/plot/clustering.py

@@ -25,6 +25,7 @@
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 """
+
 import time
 
 import matplotlib.pyplot as plt

src/sklearn_evaluation/plot/confusion_matrix_interactive.py

@@ -1,6 +1,7 @@
 """
 Plotting functions for classifier models
 """
+
 import json
 import random
 import warnings

src/sklearn_evaluation/plot/grid_search.py

@@ -1,6 +1,7 @@
 """
 Functions for visualizing grid search results
 """
+
 import collections
 
 import matplotlib.pyplot as plt

src/sklearn_evaluation/plot/matplotlib/heatmap.py

@@ -1,6 +1,7 @@
 """
 Plotting 2 discrete parameters in a heatmap
 """
+
 import numpy as np
 from sklearn_evaluation.plot.util import set_default_ax
 from sklearn_evaluation.plot.matplotlib.data_grid import DataGrid

src/sklearn_evaluation/preprocessing/binarize.py

@@ -1,6 +1,7 @@
 """
 Tools for binarizing scores
 """
+
 import numpy as np
 
 from sklearn_evaluation import validate

src/sklearn_evaluation/training/__init__.py

@@ -1,6 +1,7 @@
 """
 Utilities for model training
 """
+
 from sklearn_evaluation.training.selector import DataSelector
 
 __all__ = ["DataSelector"]

src/sklearn_evaluation/training/selector.py

@@ -4,6 +4,7 @@
 a series of transformations on your data so you can succinctly define a
 subsetting pipeline as a series of dictionaries.
 """
+
 from copy import copy, deepcopy
 import abc
 import inspect

tests/test_abstract_plot.py

@@ -1,6 +1,7 @@
 """
 Tests for AbstractPlot class
 """
+
 import pytest
 
 from sklearn_evaluation.plot.plot import AbstractPlot

tests/test_calibration_curve.py

@@ -1,6 +1,7 @@
 """
 Tests for calibration curve
 """
+
 import pytest
 import numpy as np
 from matplotlib.testing.decorators import image_comparison

tests/test_precision_recall.py

@@ -1,6 +1,7 @@
 """
 Tests for precision recall plot.
 """
+
 import pytest
 import numpy as np