Update pyright to 1.110 (#19)

Author: dbrattli

.pre-commit-config.yaml

@@ -26,5 +26,5 @@ repos:
         language: node
         pass_filenames: false
         types: [python]
-        additional_dependencies: ["[email protected].104"]
+        additional_dependencies: ["[email protected].110"]
     repo: local

expression/collections/seq.py

@@ -230,6 +230,23 @@ def scan(self, scanner: Callable[[TState, TSource], TState], state: TState) -> I
         """
         return Seq(itertools.accumulate(self, scanner, initial=state))  # type: ignore
 
+    def skip(self, count: int) -> Seq[TSource]:
+        """Returns a sequence that skips N elements of the underlying
+        sequence and then yields the remaining elements of the sequence.
+
+        Args:
+            count: The number of items to skip.
+        """
+        return Seq(pipe(self, skip(count)))
+
+    def take(self, count: int) -> Seq[TSource]:
+        """Returns the first N elements of the sequence.
+
+        Args:
+            count: The number of items to take.
+        """
+        return Seq(pipe(self, take(count)))
+
     @classmethod
     def unfold(cls, generator: Callable[[TState], Option[Tuple[TSource, TState]]], state: TState) -> Iterable[TSource]:
         """Returns a list that contains the elements generated by the
@@ -677,6 +694,20 @@ def singleton(item: TSource) -> Seq[TSource]:
     return Seq([item])
 
 
+def skip(count: int) -> Projection[Any, Any]:
+    """Returns a sequence that skips N elements of the underlying
+    sequence and then yields the remaining elements of the sequence.
+
+    Args:
+        count: The number of items to skip.
+    """
+
+    def _skip(source: Iterable[TSource]) -> Iterable[TSource]:
+        return (n for i, n in enumerate(source) if i >= count)
+
+    return _skip
+
+
 def take(count: int) -> Projection[Any, Any]:
     """Returns the first N elements of the sequence.
 
@@ -688,18 +719,7 @@ def take(count: int) -> Projection[Any, Any]:
     """
 
     def _take(source: Iterable[TSource]) -> Iterable[TSource]:
-        n = count
-
-        def gen():
-            nonlocal n
-            for x in source:
-                if n > 0:
-                    yield x
-                    n -= 1
-                else:
-                    break
-
-        return gen()
+        return (n for i, n in enumerate(source) if i < count)
 
     return _take
 
@@ -794,6 +814,7 @@ def _zip(source2: Iterable[TResult]) -> Iterable[Tuple[TSource, TResult]]:
     "of_iterable",
     "range",
     "scan",
+    "skip",
     "singleton",
     "take",
     "Projection",

tests/test_seq.py

@@ -188,6 +188,16 @@ def test_seq_collect(xs: List[int]):
     assert list(xs) == list(ys)
 
 
+@given(st.lists(st.integers()), st.integers(min_value=0))
+def test_seq_skip(xs: List[int], x: int):
+    ys = seq.of_iterable(xs)
+    try:
+        zs = pipe(ys, seq.skip(x))
+        assert list(zs) == xs[x:]
+    except ValueError:
+        assert x > len(xs)
+
+
 @given(st.lists(st.integers()), st.integers(min_value=0))
 def test_seq_take(xs: List[int], x: int):
     ys = seq.of_iterable(xs)