Skip to content

Commit 6134667

Browse files
committed
Fix trail color
1 parent 76d1701 commit 6134667

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/napari_deeplabcut/_widgets.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,13 +423,17 @@ def _show_trails(self, state):
423423
store = list(self._stores.values())[0]
424424
inds = encode_categories(store.layer.properties["label"])
425425
temp = np.c_[inds, store.layer.data]
426+
cmap = "viridis"
427+
for layer in self.viewer.layers:
428+
if isinstance(layer, Points) and layer.metadata:
429+
cmap = layer.metadata["colormap_name"]
426430
self._trails = self.viewer.add_tracks(
427431
temp,
428432
tail_length=50,
429433
head_length=50,
430434
tail_width=6,
431435
name="trails",
432-
colormap="viridis",
436+
colormap=cmap,
433437
)
434438
self._trails.visible = True
435439
elif self._trails is not None:

0 commit comments

Comments
 (0)