make load work with newer pandas (fix #34)

mdeff · Jun 17, 2020 · e8289b9 · e8289b9
1 parent 7473693
commit e8289b9
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/utils.py b/utils.py
@@ -209,8 +209,13 @@ def load(filepath):
             tracks[column] = pd.to_datetime(tracks[column])
 
         SUBSETS = ('small', 'medium', 'large')
-        tracks['set', 'subset'] = tracks['set', 'subset'].astype(
-                'category', categories=SUBSETS, ordered=True)
+        try:
+            tracks['set', 'subset'] = tracks['set', 'subset'].astype(
+                    'category', categories=SUBSETS, ordered=True)
+        except ValueError:
+            # the categories and ordered arguments were removed in pandas 0.25
+            tracks['set', 'subset'] = tracks['set', 'subset'].astype(
+                     pd.CategoricalDtype(categories=SUBSETS, ordered=True))
 
         COLUMNS = [('track', 'genre_top'), ('track', 'license'),
                    ('album', 'type'), ('album', 'information'),