LUNA16 prepare script - indexing pandas DF with a set is deprecated - turned into a list

49851c81 · Tomasz Bartczak · 98eb97f1 · 49851c81
Commit 49851c81 authored Jun 09, 2023 by Tomasz Bartczak
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

projects/Task016_Luna/scripts/prepare.py projects/Task016_Luna/scripts/prepare.py +1 -1

No files found.
--- a/projects/Task016_Luna/scripts/prepare.py
+++ b/projects/Task016_Luna/scripts/prepare.py
@@ -38,7 +38,7 @@ def create_masks(source: Path, target: Path, df: pd.DataFrame, num_processes: in
        c = []
        r = []
        try:
-            series_df = df.loc[{f.name.rsplit('.', 1)[0]}]
+            series_df = df.loc[[f.name.rsplit('.', 1)[0]]]
        except KeyError:
            pass
        else: