Remove deepcopy step during parameterize_notebook. (#673)

nteract · Aug 12, 2022 · d699e2a · d699e2a
1 parent a3f530e
commit d699e2a
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 9 deletions.
diff --git a/papermill/execute.py b/papermill/execute.py
@@ -1,6 +1,5 @@
 # -*- coding: utf-8 -*-
 
-import copy
 import nbformat
 from pathlib import Path
 
@@ -148,9 +147,6 @@ def prepare_notebook_metadata(nb, input_path, output_path, report_mode=False):
     report_mode : bool, optional
        Flag to set report mode
     """
-    # Copy the nb object to avoid polluting the input
-    nb = copy.deepcopy(nb)
-
     # Hide input if report-mode is set to True.
     if report_mode:
         for cell in nb.cells:
@@ -185,7 +181,6 @@ def prepare_notebook_metadata(nb, input_path, output_path, report_mode=False):
 
 
 def remove_error_markers(nb):
-    nb = copy.deepcopy(nb)
     nb.cells = [cell for cell in nb.cells if ERROR_MARKER_TAG not in cell.metadata.get("tags", [])]
     return nb
 

diff --git a/papermill/parameterize.py b/papermill/parameterize.py
@@ -1,4 +1,3 @@
-import copy
 import nbformat
 
 from .engines import papermill_engines
@@ -83,9 +82,6 @@ def parameterize_notebook(
     if isinstance(parameters, str):
         parameters = read_yaml_file(parameters)
 
-    # Copy the nb object to avoid polluting the input
-    nb = copy.deepcopy(nb)
-
     # Fetch out the name and language from the notebook document by dropping-down into the engine's implementation
     kernel_name = papermill_engines.nb_kernel_name(engine_name, nb, kernel_name)
     language = papermill_engines.nb_language(engine_name, nb, language)