calkit-python 0.8.2__tar.gz → 0.8.3__tar.gz

{calkit_python-0.8.2 → calkit_python-0.8.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: calkit-python
-Version: 0.8.2
+Version: 0.8.3
 Summary: Reproducibility simplified.
 Project-URL: Homepage, https://github.com/calkit/calkit
 Project-URL: Issues, https://github.com/calkit/calkit/issues

{calkit_python-0.8.2 → calkit_python-0.8.3}/calkit/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.8.2"
+__version__ = "0.8.3"
 
 from .core import *
 from . import git

{calkit_python-0.8.2 → calkit_python-0.8.3}/calkit/magics.py

@@ -29,6 +29,15 @@ class Calkit(Magics):
         required=True,
         type=_parse_string_arg,
     )
+    @magic_arguments.argument(
+        "--env",
+        nargs="?",
+        const=True,
+        help=(
+            "Whether or not this cell should be run in an environment. "
+            "If no environment name is provided, the default will be used."
+        ),
+    )
     @magic_arguments.argument(
         "--dep",
         "-d",
@@ -106,7 +115,15 @@ class Calkit(Magics):
         Then variables will be loaded back into the user namespace state by
         loading the DVC output.
         """
-        args = magic_arguments.parse_argstring(self.stage, line)
+        # Allow command for this line to continue with backslashes
+        cell_split = cell.split("\n")
+        line_combined = line.strip().removesuffix("\\")
+        while line.strip().endswith("\\"):
+            newline = cell_split.pop(0)
+            line += newline
+            line_combined += " " + newline.strip().removesuffix("\\")
+        cell = "\n".join(cell_split)
+        args = magic_arguments.parse_argstring(self.stage, line_combined)
         # If an output object type is specified, make sure we only have one
         # output
         if args.out_type:
@@ -240,7 +257,13 @@ class Calkit(Magics):
         if args.out_path:
             for path in args.out_path:
                 cmd += ["-o", path]
-        cmd.append(f"python '{script_fpath}'")
+        stage_cmd = f"python '{script_fpath}'"
+        if args.env:
+            runenv = "calkit runenv"
+            if isinstance(args.env, str):
+                runenv += f" -n {args.env}"
+            stage_cmd = runenv + " -- " + stage_cmd
+        cmd.append(stage_cmd)
         try:
             subprocess.run(cmd, check=True, capture_output=True, text=True)
         except subprocess.CalledProcessError as e:

{calkit_python-0.8.2 → calkit_python-0.8.3}/docs/tutorials/notebook-pipeline.md

@@ -143,7 +143,14 @@ We can add a line that saves the figure and declare an additional output path
 and metadata like (note this requires `plotly` and `kaleido` to be installed):
 
 ```python
-%%stage --name plot --dep get-data:df:parquet:pandas --out fig --out-path figures/plot.png --out-type figure --out-title "A plot of the data" --out-desc "This is a plot of the data."
+%%stage \
+   --name plot \
+   --dep get-data:df:parquet:pandas \
+   --out fig \
+   --out-path figures/plot.png \
+   --out-type figure \
+   --out-title "A plot of the data" \
+   --out-desc "This is a plot of the data."
 
 import os
 

{calkit_python-0.8.2 → calkit_python-0.8.3}/test/pipeline.ipynb

@@ -65,7 +65,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%%stage --name plot-fig --dep get-data:df:parquet:pandas --out fig --out-path figures/plot.png --out-type figure --out-title \"A plot of the data\" --out-desc \"This is a plot of the data.\"\n",
+    "%%stage \\\n",
+    "    --name plot-fig \\\n",
+    "    --dep get-data:df:parquet:pandas \\\n",
+    "    --out fig \\\n",
+    "    --out-path figures/plot.png \\\n",
+    "    --out-type figure \\\n",
+    "    --out-title \"A plot of the data\" \\\n",
+    "    --out-desc \"This is a plot of the data.\"\n",
     "\n",
     "import os\n",
     "\n",