From fffa0166450af844e98b0996434105fd33c8cf43 Mon Sep 17 00:00:00 2001 From: Kartik Ohri Date: Mon, 29 Apr 2024 17:05:25 +0530 Subject: [PATCH] Fix troi.patch imports --- troi/patches/ai_jams.py | 4 ++-- troi/patches/lb_radio_classes/playlist.py | 1 + troi/patches/periodic_jams_local.py | 4 ++-- troi/patches/playlist_from_listenbrainz.py | 1 - troi/patches/playlist_from_mbids.py | 5 +++-- troi/patches/recs_to_playlist.py | 5 +++-- 6 files changed, 11 insertions(+), 9 deletions(-) diff --git a/troi/patches/ai_jams.py b/troi/patches/ai_jams.py index 41af65c8..e1134d7d 100755 --- a/troi/patches/ai_jams.py +++ b/troi/patches/ai_jams.py @@ -1,11 +1,11 @@ import troi.external.gpt import troi.musicbrainz.mbid_mapping -import troi.patch from troi import Playlist +from troi.patch import Patch from troi.playlist import PlaylistMakerElement -class AiJamsPatch(troi.patch.Patch): +class AiJamsPatch(Patch): """ Generate a playlist using AI from the given prompt. """ @staticmethod diff --git a/troi/patches/lb_radio_classes/playlist.py b/troi/patches/lb_radio_classes/playlist.py index abe6f078..56e350b4 100755 --- a/troi/patches/lb_radio_classes/playlist.py +++ b/troi/patches/lb_radio_classes/playlist.py @@ -7,6 +7,7 @@ from troi import Recording from troi import TARGET_NUMBER_OF_RECORDINGS + class LBRadioPlaylistRecordingElement(troi.Element): """ Given an LB playlist, fetch its tracks and randomly include recordiungs from it. mode does not diff --git a/troi/patches/periodic_jams_local.py b/troi/patches/periodic_jams_local.py index cc7746ea..9d26ae6b 100755 --- a/troi/patches/periodic_jams_local.py +++ b/troi/patches/periodic_jams_local.py @@ -3,10 +3,10 @@ import troi.listenbrainz.recs import troi.musicbrainz.recording_lookup from troi import Playlist +from troi.patch import Patch from troi.playlist import PlaylistMakerElement from troi.local.recording_resolver import RecordingResolverElement -from troi.content_resolver.model.database import db DAYS_OF_RECENT_LISTENS_TO_EXCLUDE = 60 # Exclude tracks listened in last X days from the daily jams playlist DAILY_JAMS_MIN_RECORDINGS = 25 # the minimum number of recordings we aspire to have in a daily jam, this is not a hard limit @@ -14,7 +14,7 @@ MAX_RECS_LIMIT = 1000 # the maximum of recommendations available in LB -class PeriodicJamsLocalPatch(troi.patch.Patch): +class PeriodicJamsLocalPatch(Patch): """ """ diff --git a/troi/patches/playlist_from_listenbrainz.py b/troi/patches/playlist_from_listenbrainz.py index 3b2841f8..1fd8489b 100755 --- a/troi/patches/playlist_from_listenbrainz.py +++ b/troi/patches/playlist_from_listenbrainz.py @@ -3,7 +3,6 @@ from troi import Playlist from troi.patch import Patch from troi.playlist import PlaylistFromJSPFElement -import troi.musicbrainz.recording_lookup class TransferPlaylistPatch(Patch): diff --git a/troi/patches/playlist_from_mbids.py b/troi/patches/playlist_from_mbids.py index 0deac2ce..f633a5bf 100755 --- a/troi/patches/playlist_from_mbids.py +++ b/troi/patches/playlist_from_mbids.py @@ -2,15 +2,16 @@ from troi import Recording from troi.musicbrainz.mbid_reader import MBIDReaderElement from troi.musicbrainz.recording_lookup import RecordingLookupElement +from troi.patch import Patch from troi.playlist import PlaylistMakerElement -class PlaylistFromMBIDsPatch(troi.patch.Patch): +class PlaylistFromMBIDsPatch(Patch): """ """ def __init__(self, args): - troi.patch.Patch.__init__(self, args) + super().__init__(args) @staticmethod def inputs(): diff --git a/troi/patches/recs_to_playlist.py b/troi/patches/recs_to_playlist.py index 70709f92..8cc7cb4e 100755 --- a/troi/patches/recs_to_playlist.py +++ b/troi/patches/recs_to_playlist.py @@ -5,6 +5,7 @@ import troi.sorts import troi.musicbrainz.recording_lookup import troi.musicbrainz.mbid_mapping +from troi.patch import Patch class RecsPlaylistMakerElement(Element): @@ -53,13 +54,13 @@ def read(self, inputs): user_name=self.user_name)] -class RecommendationsToPlaylistPatch(troi.patch.Patch): +class RecommendationsToPlaylistPatch(Patch): """ See below for description """ def __init__(self, args): - troi.patch.Patch.__init__(self, args) + super().__init__(args) @staticmethod def inputs():