From 49d9e5b7ec751b82e2510f0844338ae0ea7ed299 Mon Sep 17 00:00:00 2001
From: Andi <andi.gerken@gmail.com>
Date: Wed, 19 Oct 2022 14:43:20 +0200
Subject: [PATCH] bugfix when plotting files

---
 src/robofish/io/file.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/robofish/io/file.py b/src/robofish/io/file.py
index 08dabb9..e07a252 100644
--- a/src/robofish/io/file.py
+++ b/src/robofish/io/file.py
@@ -927,6 +927,8 @@ class File(h5py.File):
                 this_c = cmap(fish_id)
             elif isinstance(c, list):
                 this_c = c[fish_id]
+            else:
+                this_c = c
 
             timesteps = poses.shape[1] - 1
             for t in range(0, timesteps, step_size):
-- 
GitLab