Skip to content

Commit c313287

Browse files
fix: airflow utils import
1 parent 017b04c commit c313287

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_airflow_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,9 @@
1414

1515
import os
1616

17-
import airflow
1817
import pytest
1918
import numpy as np
19+
from airflow import utils
2020
from airflow import DAG
2121
from airflow.contrib.operators.sagemaker_training_operator import SageMakerTrainingOperator
2222
from airflow.contrib.operators.sagemaker_transform_operator import SageMakerTransformOperator
@@ -624,7 +624,7 @@ def _build_airflow_workflow(estimator, instance_type, inputs=None, mini_batch_si
624624

625625
default_args = {
626626
"owner": "airflow",
627-
"start_date": airflow.utils.dates.days_ago(2),
627+
"start_date": utils.dates.days_ago(2),
628628
"provide_context": True,
629629
}
630630

0 commit comments

Comments
 (0)