Skip to content

Commit

Permalink
Bump flytekit to 0.11.6 and fix notebook path (#31)
Browse files Browse the repository at this point in the history
  • Loading branch information
wild-endeavor authored Aug 16, 2020
1 parent 4e81788 commit 02f4534
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 12 deletions.
20 changes: 9 additions & 11 deletions cookbook/recipes/notebook_tasks/spark_notebook_wf.py
Original file line number Diff line number Diff line change
@@ -1,19 +1,17 @@
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import os

from flytekit.sdk.types import Types
from flytekit.contrib.notebook.tasks import spark_notebook
from flytekit.sdk.tasks import inputs, outputs

from flytekit.sdk.types import Types
from flytekit.sdk.workflow import workflow_class, Input
from flytekit.contrib.notebook.tasks import spark_notebook
from flytekit.models.task import Container, Resources


interactive_spark = spark_notebook(notebook_path="./notebook-task-examples/spark-notebook.ipynb",
inputs=inputs(partitions=Types.Integer),
outputs=outputs(pi=Types.Float),
)
interactive_spark = spark_notebook(
notebook_path=os.path.join(os.path.dirname(os.path.abspath(__file__)), "spark-notebook.ipynb"),
inputs=inputs(partitions=Types.Integer),
outputs=outputs(pi=Types.Float),
)


@workflow_class
class FlyteNotebookSparkWorkflow(object):
Expand Down
2 changes: 1 addition & 1 deletion cookbook/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
flytekit[sidecar,schema]==0.10.4
flytekit[sidecar,schema]==0.11.6
pytest
statsd
opencv-python==3.4.4.19
Expand Down

0 comments on commit 02f4534

Please sign in to comment.