fixed naming bug in apple watch

DigitalBiomarkerDiscoveryPipeline · Jan 8, 2024 · 3480576 · 3480576
1 parent 1186cee
commit 3480576
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dbdpy/apple_watch.py b/dbdpy/apple_watch.py
@@ -61,7 +61,7 @@ def read_file(cls, filepath: str, device_brand: str="Apple Watch"):
         record_df["sourceName"] = record_df["sourceName"].apply(
             lambda x: unicodedata.normalize("NFKC", x)
         )
-        record_df = record_df[record_df["sourceName"].str.contains(device_name)]
+        record_df = record_df[record_df["sourceName"].str.contains(device_brand)]
 
         # Drop columns that will not be used
         drop_cols = ["sourceName", "sourceVersion", "unit", "creationDate", "device"]