From 6ea8fd46e8c35d1c74f9d443d8953fbd9473c754 Mon Sep 17 00:00:00 2001 From: Shan Carter Date: Tue, 2 Jul 2019 10:40:17 -0700 Subject: [PATCH] aligned_umap() only returns one value Fixing error when trying to unpack return values. The function only returns one value. --- lucid/recipes/activation_atlas/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lucid/recipes/activation_atlas/main.py b/lucid/recipes/activation_atlas/main.py index 03ffaa08..7e70f4a0 100644 --- a/lucid/recipes/activation_atlas/main.py +++ b/lucid/recipes/activation_atlas/main.py @@ -39,7 +39,7 @@ def activation_atlas( """Renders an Activation Atlas of the given model's layer.""" activations = layer.activations[:number_activations, ...] - layout, = aligned_umap(activations, verbose=verbose) + layout = aligned_umap(activations, verbose=verbose) directions, coordinates, _ = bin_laid_out_activations( layout, activations, grid_size )