improve code coverage

neuroinformatics-unit · Feb 6, 2025 · 7701aa3 · 7701aa3
1 parent 69a0a6e
commit 7701aa3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/movement/plot.py b/movement/plot.py
@@ -105,11 +105,10 @@ def trajectory(
     for key, value in DEFAULT_PLOTTING_ARGS.items():
         kwargs.setdefault(key, value)
 
+    colorbar = False
     if "c" not in kwargs:
         kwargs["c"] = plot_point.time
         colorbar = True
-    else:
-        colorbar = False
 
     # Plot the scatter, coloring by time or user-provided color
     sc = ax.scatter(