sunank200 commented on code in PR #53387:
URL: https://github.com/apache/airflow/pull/53387#discussion_r2215592101


##########
airflow-core/src/airflow/api_fastapi/core_api/datamodels/ui/calendar.py:
##########
@@ -21,7 +21,12 @@
 
 from pydantic import BaseModel
 
-from airflow.utils.state import DagRunState
+from airflow.version_compat import AIRFLOW_V_3_1_PLUS
+
+if AIRFLOW_V_3_1_PLUS:
+    from airflow.sdk.definitions.state import DagRunState
+else:
+    from airflow.utils.state import DagRunState  # type: ignore[no-redef]

Review Comment:
   That's right now when I see it. I will change this. My thought process was 
that `DagRunState` is also used in providers, and I missed that the core will 
always be on latest version.
   
   After discussion with Amogh, the state is a good candidate for 
`airflow-shared`



-- 
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