Update src/phoenix/pcloud/projectors.py

Co-authored-by: Mikyo King <mikyo@arize.com>
Arize-ai · Nov 23, 2022 · 0970ab3 · 0970ab3
1 parent e29c344
commit 0970ab3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/phoenix/pcloud/projectors.py b/src/phoenix/pcloud/projectors.py
@@ -55,7 +55,7 @@ def project(self, primary_dataset: Dataset, reference_dataset: Dataset, embeddin
         # Sample down our datasets to max 2500 rows for UMAP performance
         points_per_dataset = MAX_UMAP_POINTS // 2
         sampled_primary_dataset = primary_dataset.sample(
-            num=MAX_UMAP_POINTS // 2,
+            num=points_per_dataset
         )
         sampled_reference_dataset = reference_dataset.sample(num=MAX_UMAP_POINTS // 2)