Fixes integer columns for read_fit()

GoldenCheetah · Apr 20, 2021 · 8668735 · 8668735
1 parent 9d8432d
commit 8668735
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/docs/docs/changelog.md b/docs/docs/changelog.md
@@ -16,6 +16,8 @@ Types of changes:
 - `Security` in case of vulnerabilities.
 
 ## [Unreleased] - {ISO8601 date}
+### Fixed
+- `sweat.read_fit()` now supports integer column names.
 
 ## [0.20.2] - 2021-04-16
 ### Fixed

diff --git a/sweat/io/fit.py b/sweat/io/fit.py
@@ -18,7 +18,7 @@
 
 def process_location_columns(df, columns=None):
     if columns is None:
-        columns = [c for c in df.columns if c.endswith("_lat") or c.endswith("_long")]
+        columns = [c for c in df.columns if isinstance(c, str) and (c.endswith("_lat") or c.endswith("_long"))]
     else:
         columns = [c for c in columns if c in df.columns]