blob: 925ba00e2b1c97fde45824ad80139820e5961db0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from collections.abc import Callable
from typing import Any, Iterable, TypeVar, cast
T = TypeVar("T")
D = TypeVar("D")
def remove_element(
iterable: Iterable[T | None], to_rm: Iterable[Any], des: type[D] | None = None
) -> D:
to_rm = set(to_rm)
return cast(Callable[..., D], des or list)(v for v in iterable if v not in to_rm)
def remove_none(iterable: Iterable[T | None], des: type[D] | None = None) -> D:
return cast(Callable[..., D], des or list)(v for v in iterable if v is not None)
|
