dstandish commented on code in PR #39604: URL: https://github.com/apache/airflow/pull/39604#discussion_r1602029456
########## 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: i think that there was actually no mypy complaint, just a ruff complaint about the docstring "mood" -- 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