Make imports flexible.

Author: tobiasraabe

src/pytask_parallel/execute.py

@@ -11,9 +11,6 @@
 from typing import Callable
 
 import cloudpickle
-from _pytask.warnings import parse_warning_filter
-from _pytask.warnings import warning_record_to_str
-from _pytask.warnings_utils import WarningReport
 from pybaum.tree_util import tree_map
 from pytask import console
 from pytask import ExecutionReport
@@ -27,6 +24,16 @@
 from rich.console import ConsoleOptions
 from rich.traceback import Traceback
 
+# Can be removed if pinned to pytask >= 0.2.6.
+try:
+    from pytask import parse_warning_filter
+    from pytask import warning_record_to_str
+    from pytask import WarningReport
+except ImportError:
+    from _pytask.warnings import parse_warning_filter
+    from _pytask.warnings import warning_record_to_str
+    from _pytask.warnings_utils import WarningReport
+
 
 @hookimpl
 def pytask_post_parse(config: dict[str, Any]) -> None: