From ec2f2d734fc92401ca23aae2628fa119d245d3a0 Mon Sep 17 00:00:00 2001 From: cloudwithax Date: Sun, 13 Nov 2022 22:01:15 -0500 Subject: [PATCH] make queue.remove take Track instead of int --- pomice/queue.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pomice/queue.py b/pomice/queue.py index a44a8aa..97d0cf7 100644 --- a/pomice/queue.py +++ b/pomice/queue.py @@ -118,15 +118,14 @@ class Queue(Iterable[Track]): def _index(self, item: Track) -> int: return self._queue.index(item) - def _put(self, item: Track) -> None: self._queue.append(item) def _insert(self, index: int, item: Track) -> None: self._queue.insert(index, item) - def _remove(self, index: int) -> None: - self._queue.remove(index) + def _remove(self, item: Track) -> None: + self._queue.remove(item) def _get_random_float(self) -> float: return random.random() @@ -233,7 +232,7 @@ class Queue(Iterable[Track]): Removes a item within the queue. Raises ValueError if item is not in queue. """ - return self._remove(self._index(self._check_track(item))) + return self._remove(self._check_track(item)) def find_position(self, item: Track) -> int: