Fix quantized_embedding_norm undefined when normalize=False

CeliaBenquet · CeliaBenquet · commit c7e0e65eb9ee · 2025-04-25T18:33:26.000+02:00
diff --git a/cebra/integrations/sklearn/helpers.py b/cebra/integrations/sklearn/helpers.py
@@ -155,6 +155,8 @@ def align_embeddings(
                 quantized_sample / np.linalg.norm(quantized_sample, axis=0)
                 for quantized_sample in quantized_embedding
             ]
+            quantized_embeddings.append(quantized_embedding_norm)
+        else:
+            quantized_embeddings.append(quantized_embedding)
 
-        quantized_embeddings.append(quantized_embedding_norm)
     return quantized_embeddings

Original file line number	Diff line number	Diff line change
`@@ -155,6 +155,8 @@ def align_embeddings(`
`155`	`155`	`quantized_sample / np.linalg.norm(quantized_sample, axis=0)`
`156`	`156`	`for quantized_sample in quantized_embedding`
`157`	`157`	`]`
	`158`	`+ quantized_embeddings.append(quantized_embedding_norm)`
	`159`	`+ else:`
	`160`	`+ quantized_embeddings.append(quantized_embedding)`
`158`	`161`
`159`		`- quantized_embeddings.append(quantized_embedding_norm)`
`160`	`162`	`return quantized_embeddings`