diff --git a/api/funkwhale_api/radios/radios.py b/api/funkwhale_api/radios/radios.py index e4e99b45b..93825c03a 100644 --- a/api/funkwhale_api/radios/radios.py +++ b/api/funkwhale_api/radios/radios.py @@ -184,9 +184,6 @@ class SessionRadio(SimpleRadio): return sliced_queryset[:quantity] - def pick_v2(self, **kwargs): - return self.pick_many_v2(quantity=1, **kwargs)[0] - def pick_many_v2(self, quantity, **kwargs): if self.session: sliced_queryset = self.get_choices_v2(quantity=quantity, **kwargs) diff --git a/api/tests/radios/test_radios.py b/api/tests/radios/test_radios.py index b82321906..6760fb3e0 100644 --- a/api/tests/radios/test_radios.py +++ b/api/tests/radios/test_radios.py @@ -525,7 +525,7 @@ def test_can_cache_radio_track(factories): user = factories["users.User"]() radio = radios.RandomRadio() session = radio.start_session(user) - picked = session.radio.pick_v2(filter_playable=False) + picked = session.radio.pick_many_v2(quantity=1, filter_playable=False) assert len(picked) == 1 for t in cache.get(f"radiosessiontracks{session.id}"): assert t.track in uploads