Add missing dimensionality reduction method specification in a test

phurwicz · Apr 27, 2023 · ca59774 · ca59774
1 parent d6b1172
commit ca59774
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/core/representation/test_reduction.py b/tests/core/representation/test_reduction.py
@@ -28,7 +28,7 @@ def test_dimensionality_reduction(n_points=1000):
     reducer.fit_transform(
         "umap", n_neighbors=3, min_dist=0.01, dimension=3, metric="euclidean"
     )
-    embedding = reducer.transform(arr)
+    embedding = reducer.transform(arr, "umap")
     assert embedding.shape == (n_points, 3)
     embedding = reducer.transform(np.array([]))
     assert embedding.shape == (0,)