add ploomber-extension

CHANGELOG.md

@@ -1,7 +1,9 @@
 # CHANGELOG
 
 ## 0.0.32dev
+
 * [Feature] Add benchmarking CLI `python -m ploomber_engine.benchmark`
+* [Feature] Add `ploomber-extension` as a dependency
 
 ## 0.0.31 (2023-08-24)
 

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

@@ -17,6 +17,7 @@
 
 REQUIRES = [
     "ploomber-core>=0.2",
+    "ploomber-extension",
     "debuglater>=1.4.4",
     # used for our debug now feature (PloomberNotebookClient)
     "nbclient",
@@ -37,7 +38,8 @@
 
 DEV = [
     "pkgmt",
-    "pytest",
+    # using pytest 7 because of an issue when using @pytest.mark.memory
+    "pytest<8",
     "flake8",
     "invoke",
     "twine",

src/ploomber_engine/benchmark.py

@@ -1,6 +1,7 @@
 """
 A module to benchmark notebooks in a directory.
 """
+
 from pathlib import Path
 
 import click

src/ploomber_engine/cli.py

@@ -1,6 +1,7 @@
 """
 Command-line interface
 """
+
 import ast
 import click
 

src/ploomber_engine/execute.py

@@ -1,6 +1,7 @@
 """
 Abstractions for running notebooks with papermill-like interface
 """
+
 import warnings
 from pathlib import Path
 import csv

src/ploomber_engine/ipython.py

@@ -489,9 +489,11 @@ def execute(self, parameters=None):
         if self._debug_later:
             add_debuglater_cells(
                 self._nb,
-                path_to_dump=self._debug_later
-                if isinstance(self._debug_later, (str, Path))
-                else None,
+                path_to_dump=(
+                    self._debug_later
+                    if isinstance(self._debug_later, (str, Path))
+                    else None
+                ),
             )
 
         with self:

src/ploomber_engine/testing.py

@@ -1,6 +1,7 @@
 """
 Testing notebooks against their recorded outputs
 """
+
 import warnings
 
 import nbformat

src/ploomber_engine/tracking/io.py

@@ -2,6 +2,7 @@
 NOTE: this was copied from
 https://github.com/ploomber/jupyblog/blob/master/src/jupyblog/execute.py
 """
+
 from pathlib import Path
 import base64
 import re

tests/test_cli.py

@@ -1,6 +1,7 @@
 """
 Command-line interface tests
 """
+
 from unittest.mock import Mock, call
 
 import pytest

tests/test_tracking.py

@@ -311,4 +311,4 @@ def test_tracking_import_telemetry(tmp_empty, monkeypatch):
         quiet=True,
     )
 
-    assert mock.call_count == 2
+    assert mock.call_count == 1