diff --git a/api/schedule/clean_dataset_queries_task.py b/api/schedule/clean_dataset_queries_task.py index aa0c72aa60..0760e89bc0 100644 --- a/api/schedule/clean_dataset_queries_task.py +++ b/api/schedule/clean_dataset_queries_task.py @@ -88,13 +88,10 @@ def clean_dataset_queries_task() -> None: except LockError: end_at = time.perf_counter() - logger.warning( - "clean_dataset_queries_task: lock already held, skip current execution" - ) + logger.warning("clean_dataset_queries_task: lock already held, skip current execution") click.echo( click.style( - f"clean_dataset_queries_task: skipped (lock already held), " - f"latency: {end_at - start_at:.2f}s", + f"clean_dataset_queries_task: skipped (lock already held), latency: {end_at - start_at:.2f}s", fg="yellow", ) ) diff --git a/api/tests/unit_tests/schedule/test_clean_dataset_queries_task.py b/api/tests/unit_tests/schedule/test_clean_dataset_queries_task.py index 792bc070e0..458110e041 100644 --- a/api/tests/unit_tests/schedule/test_clean_dataset_queries_task.py +++ b/api/tests/unit_tests/schedule/test_clean_dataset_queries_task.py @@ -1,6 +1,5 @@ from unittest.mock import MagicMock, patch -import pytest from redis.exceptions import LockError from schedule.clean_dataset_queries_task import _effective_retention_days, clean_dataset_queries_task