@@ -54,7 +54,7 @@ def convert_notebook_to_script(
5454 if clear_output :
5555 util .clear_directory (output_dir )
5656 exporter = nbconvert .PythonExporter ()
57- ( body , resources ) = exporter .from_notebook_node (self .notebook )
57+ body , resources = exporter .from_notebook_node (self .notebook )
5858 with open (output_dir / "user_code.py" , "w" ) as fh :
5959 fh .write (
6060 "import unittest.mock\n "
@@ -84,7 +84,7 @@ def process_params_cell(self) -> None:
8484 setup_node = nbformat .from_dict (self .notebook )
8585 setup_node .cells = setup_node .cells [:params_cell_index ]
8686 exporter = nbconvert .PythonExporter ()
87- ( setup_code , _ ) = exporter .from_notebook_node (setup_node )
87+ setup_code , _ = exporter .from_notebook_node (setup_node )
8888 # Mock out the get_ipython function in case there are any IPython
8989 # magic commands in the notebook. This effectively turns them into
9090 # no-ops.
@@ -94,7 +94,7 @@ def process_params_cell(self) -> None:
9494 )
9595 params_node = nbformat .from_dict (self .notebook )
9696 params_node .cells = [params_node .cells [params_cell_index ]]
97- ( params_code , _ ) = exporter .from_notebook_node (params_node )
97+ params_code , _ = exporter .from_notebook_node (params_node )
9898 self .nb_params = NotebookParameters .from_code (
9999 params_code ,
100100 setup_code = setup_code ,
@@ -308,8 +308,7 @@ def ensure_present(pkg: str):
308308
309309 def _build_image (self ) -> docker .models .images .Image :
310310 client = docker .from_env ()
311- dockerfile = textwrap .dedent (
312- """
311+ dockerfile = textwrap .dedent ("""
313312 FROM mambaorg/micromamba:1.5.10-noble-cuda-12.6.0
314313 COPY Dockerfile Dockerfile
315314 COPY environment.yml environment.yml
@@ -322,8 +321,7 @@ def _build_image(self) -> docker.models.images.Image:
322321 COPY parameters.py parameters.py
323322 COPY util.py util.py
324323 ENTRYPOINT ["/usr/local/bin/_entrypoint.sh", "python", "/home/mambauser/execute.py"]
325- """
326- )
324+ """ )
327325 with open (self .build_dir / "Dockerfile" , "w" ) as fh :
328326 fh .write (dockerfile )
329327 LOGGER .info (f"Building image with tag { self .tag } ..." )
0 commit comments