jedcunningham commented on code in PR #48014:
URL: https://github.com/apache/airflow/pull/48014#discussion_r2005903983


##########
airflow/example_dags/example_custom_weight.py:
##########
@@ -24,12 +24,12 @@
 import pendulum
 
 from airflow.example_dags.plugins.decreasing_priority_weight_strategy import 
DecreasingPriorityStrategy
-
-# [START example_custom_weight_dag]
-from airflow.models.dag import DAG
 from airflow.providers.standard.operators.bash import BashOperator
 from airflow.providers.standard.operators.empty import EmptyOperator
 
+# [START example_custom_weight_dag]
+from airflow.sdk import DAG

Review Comment:
   ```suggestion
   from airflow.providers.standard.operators.empty import EmptyOperator
   from airflow.sdk import DAG
   
   # [START example_custom_weight_dag]
   ```



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to