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