Pass pre-commit.

mosaicml · Dec 30, 2023 · 1ef3e5d · 1ef3e5d
1 parent a51a02c
commit 1ef3e5d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/scripts/writing/bench.py b/scripts/writing/bench.py
@@ -10,7 +10,7 @@
 from functools import partial
 from shutil import rmtree
 from time import time
-from typing import Dict, Iterable, List, Optional, Tuple, TypeVar
+from typing import Any, Dict, Iterable, List, Optional, Tuple, TypeVar
 
 import lance
 import numpy as np
@@ -259,7 +259,7 @@ def _write_csv(nums: List[int],
                txts: List[str],
                root: str,
                size_limit: Optional[int],
-               show_progress: bool = True) -> None:
+               show_progress: bool = True) -> Dict[str, Any]:
     """Save the dataset in Streaming CSV form.
 
     Args:
@@ -291,7 +291,7 @@ def _write_jsonl(nums: List[int],
                  txts: List[str],
                  root: str,
                  size_limit: Optional[int],
-                 show_progress: bool = True) -> None:
+                 show_progress: bool = True) -> Dict[str, Any]:
     """Save the dataset Streaming JSON form.
 
     Args:
@@ -323,7 +323,7 @@ def _write_mds(nums: List[int],
                txts: List[str],
                root: str,
                size_limit: Optional[int],
-               show_progress: bool = True) -> None:
+               show_progress: bool = True) -> Dict[str, Any]:
     """Save the dataset in Streaming MDS form.
 
     Args:

diff --git a/streaming/base/format/base/timer.py b/streaming/base/format/base/timer.py
@@ -206,7 +206,7 @@ def get_stats(self, num_groups: Optional[int] = 100) -> Dict[str, Any]:
                 part = timer['stats']['total']
                 timer['stats']['frac_of_parts'] = part / sum_of_parts
 
-            obj['named_timers'] = named_timers
+            obj['named_timers'] = named_timers  # pyright: ignore
 
         return obj