Skip to content

Instantly share code, notes, and snippets.

@joannapurosto

joannapurosto/airflow-Valohai-task.py

Last active November 21, 2019 07:37
Show Gist options
  • Save joannapurosto/b892e894f73e62fd064005482b67a39f to your computer and use it in GitHub Desktop.
Save joannapurosto/b892e894f73e62fd064005482b67a39f to your computer and use it in GitHub Desktop.
Download ZIP
Airflow Valohai Operator for Valohai Blog
Raw
airflow-Valohai-task.py
from airflow.operators.valohai import ValohaiSubmitExecutionOperator
train = ValohaiSubmitExecutionOperator(
task_id='train_model',
# Specify the project and step from Valohai
project_name='tensorflow-example',
step='Train model (MNIST)',
# Associate a task to a previously created DAG
dag=dag,
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment