Fix ruff issues

Author: lukasbindreiter

tilebox-datasets/tilebox/datasets/progress.py

@@ -3,6 +3,11 @@
 from types import TracebackType
 from typing import Any
 
+try:
+    from typing import Self  # ty: ignore[unresolved-import]
+except ImportError:  # Self is only available in Python 3.11+
+    from typing_extensions import Self
+
 from tqdm.auto import tqdm
 
 from tilebox.datasets.query.time_interval import TimeInterval
@@ -42,7 +47,7 @@ def __init__(
         self._actual_start_time = actual_start_time
         self._total_data_points = 0
 
-    def __enter__(self) -> "TimeIntervalProgressBar":
+    def __enter__(self) -> Self:
         self._progress_bar = tqdm(
             bar_format="{l_bar}{bar}[{elapsed}<{remaining}{postfix}]",
             total=self._calc_progress_seconds(self._interval.end),

tilebox-workflows/tilebox/workflows/runner/task_runner.py

@@ -19,6 +19,11 @@
 from uuid import UUID
 from warnings import warn
 
+try:
+    from typing import Self  # ty: ignore[unresolved-import]
+except ImportError:  # Self is only available in Python 3.11+
+    from typing_extensions import Self
+
 from loguru import logger
 from opentelemetry.trace.status import StatusCode
 from tenacity import retry, retry_if_exception_type, stop_when_event_set, wait_random_exponential
@@ -260,7 +265,7 @@ def sleep(self, seconds: float) -> None:
         """Sleep for a given number of seconds, or until an interrupt signal is received."""
         self._interrupted.wait(seconds)
 
-    def __enter__(self) -> "_GracefulShutdown":
+    def __enter__(self) -> Self:
         """Enter a graceful shutdown context. Intercepts SIGTERM and SIGINT signals and delays them by a grace period."""
         self._original_sigterm = signal.signal(signal.SIGTERM, self._external_interrupt_handler)
         self._original_sigint = signal.signal(signal.SIGINT, self._external_interrupt_handler)