kubeflow
/
pipelines
mirror of https://github.com/kubeflow/pipelines.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix an issue that %%docker doesn't work. (#119)

This commit is contained in:
qimingj 2018-11-06 20:06:58 -08:00 committed by k8s-ci-robot
parent 849c8cfe77
commit bb62b321a5
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
sdk/python/kfp/notebook/_magic.py
View File

@ -21,6 +21,7 @@ except ImportError:
from kfp.compiler import build_docker_image
import os
import tempfile
@ -35,8 +36,9 @@ def docker(line, cell):
target, staging = line.split()
with tempfile.NamedTemporaryFile(mode='wt') as f:
f.write(cell)
build_docker_image(staging, target, f.name)
with tempfile.NamedTemporaryFile(mode='wt', delete=False) as f:
f.write(cell)
build_docker_image(staging, target, f.name)
os.remove(f.name)