uranusjr commented on code in PR #39604:
URL: https://github.com/apache/airflow/pull/39604#discussion_r1600462135
##########
airflow/models/abstractoperator.py:
##########
@@ -395,6 +395,19 @@ def get_closest_mapped_task_group(self) -> MappedTaskGroup
| None:
"""
return next(self.iter_mapped_task_groups(), None)
+ def get_needs_expansion(self) -> bool:
+ """
+ Returns true of the task is MappedOperator or is in a mapped task
group.
+
+ :meta private:
+ """
+ if self._needs_expansion is None:
+ if self.get_closest_mapped_task_group() is not None:
+ self._needs_expansion = True
+ else:
+ self._needs_expansion = False
+ return self._needs_expansion
Review Comment:
```suggestion
if self._needs_expansion is not None:
return self._needs_expansion
if self.get_closest_mapped_task_group() is not None:
needs_expansion = True
else:
needs_expansion = False
self._needs_expansion = needs_expansion
return needs_expansion
```
I think this is needed to get past Mypy…
--
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: commits-unsubscr...@airflow.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
