Hotfix: Fixed wrong dataclass

OrfiTeam · Feb 6, 2022 · ab2a3ee · ab2a3ee
1 parent 81a80fd
commit ab2a3ee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/orpheus/music_downloader.py b/orpheus/music_downloader.py
@@ -260,8 +260,8 @@ def download_track(self, track_id, album_location='', main_artist='', track_inde
         album_location = album_location.replace('\\', '/')
 
         # Ignores "single_full_path_format" and just downloads every track as an album
-        if self.global_settings['formatting']['force_album_format'] and self.download_mode in [
-            DownloadTypeEnum.track, DownloadTypeEnum.playlist]:
+        if self.global_settings['formatting']['force_album_format'] and self.download_mode in {
+            DownloadTypeEnum.track, DownloadTypeEnum.playlist}:
             # Fetch every needed album_info tag and create an album_location
             album_info: AlbumInfo = self.service.get_album_info(track_info.album_id)
             # Save the playlist path to save all the albums in the playlist path

diff --git a/utils/models.py b/utils/models.py
@@ -166,7 +166,6 @@ class ExtensionInformation:
     settings: dict
 
 
-@dataclass
 class DownloadTypeEnum(Flag):
     track = auto()
     playlist = auto()