calkit-python 0.24.0__tar.gz → 0.24.1__tar.gz

{calkit_python-0.24.0 → calkit_python-0.24.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: calkit-python
-Version: 0.24.0
+Version: 0.24.1
 Summary: Reproducibility simplified.
 Project-URL: Homepage, https://calkit.org
 Project-URL: Issues, https://github.com/calkit/calkit/issues

{calkit_python-0.24.0 → calkit_python-0.24.1}/calkit/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.24.0"
+__version__ = "0.24.1"
 
 from .core import *  # noqa: F403, I001
 from . import git  # noqa: F401

{calkit_python-0.24.0 → calkit_python-0.24.1}/calkit/models/pipeline.py

@@ -343,12 +343,12 @@ class JupyterNotebookStage(Stage):
     @property
     def dvc_outs(self) -> list[str | dict]:
         outs = super().dvc_outs
-        # TODO: This should also export HTML?
         exec_nb_path = self.executed_notebook_path
-        exec_nb_out = {
-            exec_nb_path: {"cache": self.executed_ipynb_storage == "dvc"}
-        }
-        outs = outs + [exec_nb_out]
+        html_path = self.html_path
+        outs = outs + [
+            {exec_nb_path: {"cache": self.executed_ipynb_storage == "dvc"}},
+            {html_path: {"cache": self.html_storage == "dvc"}},
+        ]
         return outs
 
     @property

{calkit_python-0.24.0 → calkit_python-0.24.1}/calkit/tests/test_pipeline.py

@@ -194,9 +194,9 @@ def test_to_dvc_notebook_stage():
                     "outputs": [
                         "figures/fig1.png",
                     ],
-                    "html_storage": "git",
-                    "cleaned_ipynb_storage": None,
-                    "executed_ipynb_storage": "dvc",
+                    "html_storage": "dvc",
+                    "cleaned_ipynb_storage": "git",
+                    "executed_ipynb_storage": None,
                 },
             }
         }
@@ -208,3 +208,12 @@ def test_to_dvc_notebook_stage():
     assert clean_stage["desc"].startswith("Automatically generated")
     stage = dvc_stages["notebook-1"]
     assert "--to html" in stage["cmd"]
+    found_html = False
+    for out in stage["outs"]:
+        if not isinstance(out, dict):
+            continue
+        p = list(out.keys())[0]
+        if p.endswith(".html"):
+            found_html = True
+            assert out[p]["cache"]
+    assert found_html