Update interactive_context.py

m2kar · Jun 2, 2020 · e171fe0 · e171fe0
1 parent d8dd07b
commit e171fe0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tfx/orchestration/experimental/interactive/interactive_context.py b/tfx/orchestration/experimental/interactive/interactive_context.py
@@ -188,8 +188,8 @@ def export_to_pipeline(self, notebook_filepath: Text, export_filepath: Text,
     absl.logging.info('Exporting contents of %s to %s with %s runner.',
                       notebook_filepath, export_filepath, runner_type)
 
-    with open(notebook_filepath,encoding="utf-8") as notebook_f,\
-        open(export_filepath, 'w',encoding="utf-8") as export_f:
+    with open(notebook_filepath, encoding="utf-8") as notebook_f,\
+        open(export_filepath, 'w', encoding="utf-8") as export_f:
       notebook = nbformat.read(notebook_f, nbformat.NO_CONVERT)
       cells = notebook['cells']
       code_cells = (cell for cell in cells if cell['cell_type'] == 'code')