Improve example DAG for ML Engine (#7810)

apache · Mar 23, 2020 · a001489 · a001489
1 parent 210de87
commit a001489
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow/providers/google/cloud/example_dags/example_mlengine.py b/airflow/providers/google/cloud/example_dags/example_mlengine.py
@@ -172,7 +172,7 @@
 
  training >> create_version
  training >> create_version_2
- create_model >> get_model >> get_model_result
+ create_model >> get_model >> [get_model_result, delete_model]
  create_model >> create_version >> create_version_2 >> set_defaults_version >> list_version
  create_version >> prediction
  create_version_2 >> prediction