Skip to content

Commit 5d4d24b

Browse files
author
Jan Sebastian Rothe
committed
Added function to silence logger
1 parent 0b6e79c commit 5d4d24b

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

src/fagfunksjoner/fagfunksjoner_logger.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,13 @@
22

33
import logging
44
import sys
5+
from collections.abc import Callable
56
from typing import Any
67

78
from colorama import Back, Fore, Style
89

910

10-
def silence_logger(func: Callable, *args, **kwargs) -> Any:
11+
def silence_logger(func: Callable[..., Any], *args: Any, **kwargs: Any) -> Any:
1112
"""Silences INFO and WARNING logs for the duration of the function call."""
1213
original_level = logger.level
1314
logger.setLevel(logging.ERROR) # Suppress INFO and WARNING messages

tests/paths/test_versions.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,9 @@ def test_next_version_number(mock_get_fileversions, mock_latest_version_path):
8181
# Test for next_version_path function
8282
@patch("fagfunksjoner.paths.versions.latest_version_path")
8383
@patch("fagfunksjoner.paths.versions.next_version_number")
84-
def test_next_version_path(mock_next_version_number, mock_latest_version_path, mock_get_version_number):
84+
def test_next_version_path(
85+
mock_next_version_number, mock_latest_version_path, mock_get_version_number
86+
):
8587
mock_next_version_number.return_value = 3
8688
mock_latest_version_path.return_value = "gs://bucket/folder/file_v2.parquet"
8789
mock_get_version_number.return_value = 1

0 commit comments

Comments
 (0)