contrib: fix importing pandas datasets in excel format (#627)

Fixes #617 Co-authored-by: ahari <[email protected]>
iterative · May 11, 2023 · 5f3a480 · 5f3a480
1 parent 5e450b0
commit 5f3a480
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mlem/contrib/pandas.py b/mlem/contrib/pandas.py
@@ -270,6 +270,8 @@ def get_writer(
             ext = ext.lstrip(".")
             if ext in PANDAS_SERIES_FORMATS:
                 fmt = ext
+            elif ext in ["xls", "xlsx"]:
+                fmt = "excel"
         return PandasSeriesWriter(format=fmt)
 
 
@@ -367,6 +369,8 @@ def get_writer(
             ext = ext.lstrip(".")
             if ext in PANDAS_FORMATS:
                 fmt = ext
+            elif ext in ["xls", "xlsx"]:
+                fmt = "excel"
         return PandasWriter(format=fmt)