fix

src/chtools/chadmin/cli/chs3_backup_group.py

@@ -68,8 +68,8 @@ def cleanup_backups(ctx, dry_run, keep_going):
 
 
 def delete_chs3_backups(
-    ctx: Context, chs3_backups: List[str], *, keep_going=False, dry_run=False
-) -> None:
+    ctx, chs3_backups, *, keep_going=False, dry_run=False
+):
     """
     Delete CHS3 backups.
     """

src/chtools/chadmin/cli/object_storage_group.py

@@ -4,7 +4,7 @@
 import sys
 from datetime import datetime, timedelta, timezone
 from gzip import GzipFile
-from io import BufferedIOBase, IOBase, TextIOWrapper
+from io import TextIOWrapper, IOBase
 from pathlib import Path
 from typing import Dict, Iterator, List, Optional, Union
 
@@ -173,7 +173,7 @@ def list_objects(
     help="Input stream is compressed using GZIP format",
 )
 @pass_context
-def clean_object_storage(ctx: Context, file: IOBase, compressed: bool) -> None:
+def clean_object_storage(ctx , file , compressed ):
     disk_conf: S3DiskConfiguration = ctx.obj["disk_configuration"]
 
     if compressed:
@@ -190,13 +190,11 @@ def clean_object_storage(ctx: Context, file: IOBase, compressed: bool) -> None:
 
 @contextlib.contextmanager
 def dump_writer(
-    compressed: bool, file_path: Optional[Path] = None
-) -> Iterator[Union[BufferedIOBase, GzipFile]]:
+    compressed, file_path = None
+):
     writer = open(file_path, "wb") if file_path is not None else sys.stdout.buffer
-    if compressed:
-        writer = GzipFile(mode="wb", fileobj=writer)
     try:
-        yield writer
+        yield GzipFile(mode="wb", fileobj=writer) if compressed else writer
     finally:
         if file_path is not None or compressed:
             writer.close()

src/chtools/chadmin/internal/diagnostics/data.py

@@ -8,7 +8,7 @@
 import yaml
 from requests.exceptions import RequestException
 
-from chtools.common.clickhouse.client import OutputFormat
+from chtools.common.clickhouse.client import OutputFormat, ClickhouseClient
 
 from .utils import delayed
 
@@ -195,8 +195,8 @@ def _write_result(buffer_, result, format_=None):
 
 @delayed
 def add_query(
-    diagnostics, name, client, query, format_: OutputFormat, section=None
-) -> None:
+    diagnostics, name, client, query, format_, section=None
+):
     query_args = {
         "normalize_queries": diagnostics.normalize_queries,
     }
@@ -210,7 +210,7 @@ def add_query(
 
 
 def execute_query(
-    client: ClickHouseClient,
+    client: ClickhouseClient,
     query: str,
     render_query: bool = True,
     format_: OutputFormat = OutputFormat.Default,

src/chtools/common/clickhouse/client/clickhouse_client.py

@@ -64,11 +64,11 @@ def query(
         query: str,
         query_args: Optional[Dict[str, Any]] = None,
         format_: Optional[str] = None,
-        post_data=None,
+        post_data: Any = None,
         timeout: Optional[int] = None,
         echo: bool = False,
         dry_run: bool = False,
-    ):
+    ) -> Any:
         """
         Execute query.
         """

tests/unit/common/type/test_typed_enum.py

@@ -44,5 +44,5 @@ def test_typed_enum(
     stringified: Sequence[str] = [str(i) for i in inputs]
     assert_that(stringified, equal_to(stringified_expected))
 
-    summed = reduce(lambda a, b: a + b, inputs)
+    summed = reduce(lambda a, b: a + b, inputs)  # type: ignore
     assert_that(summed, equal_to(summed_expected))