From 5fd72d645ce609bb60076c901d508c89b50870a0 Mon Sep 17 00:00:00 2001 From: Andi <andi.gerken@gmail.com> Date: Thu, 15 Dec 2022 18:57:02 +0100 Subject: [PATCH] Fixed minibug --- src/conversion_scripts/convert_from_csv.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/conversion_scripts/convert_from_csv.py b/src/conversion_scripts/convert_from_csv.py index 790019c..def488e 100644 --- a/src/conversion_scripts/convert_from_csv.py +++ b/src/conversion_scripts/convert_from_csv.py @@ -231,6 +231,7 @@ def handle_file(file, args): supress = [] + all_switches = None if not args.disable_fix_switches: for run in range(20): print("RUN ", run) @@ -258,8 +259,8 @@ def handle_file(file, args): for f in range(n_fish): iof.create_entity("fish", poses[:, f]) - - iof.attrs["switches"] = all_switches + if all_switches is not None: + iof.attrs["switches"] = all_switches # assert np.all( # poses[np.logical_not(np.isnan(poses[:, 0])), 0] >= 0 -- GitLab