diff --git a/music_kraken/objects/collection.py b/music_kraken/objects/collection.py index a257226..d08415e 100644 --- a/music_kraken/objects/collection.py +++ b/music_kraken/objects/collection.py @@ -84,6 +84,7 @@ class Collection(Generic[T]): for e in self._data: self._map_element(e, no_unmap=True) + def _find_object(self, __object: T, **kwargs) -> Optional[T]: self._remap() @@ -216,8 +217,8 @@ class Collection(Generic[T]): self._data.remove(existing) self._unmap_element(existing) - def contains(self, other: T) -> bool: - return self._find_object(other) is not None + def contains(self, __object: T) -> bool: + return self._find_object(__object) is not None def extend(self, other_collections: Optional[Generator[T, None, None]], **kwargs): if other_collections is None: