From e3af2a2e3337464897190b8b603d14a93725b72f Mon Sep 17 00:00:00 2001 From: jacobthebanana Date: Thu, 13 Nov 2025 17:24:49 +0000 Subject: [PATCH] Fixed import errors in diversity visualization. Fixed whitespace issues in diversity visualization dataset_name. --- .../2_synthetic_data/gradio_visualize_diversity.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/3_evals/2_synthetic_data/gradio_visualize_diversity.py b/src/3_evals/2_synthetic_data/gradio_visualize_diversity.py index dfa222b..b01150f 100644 --- a/src/3_evals/2_synthetic_data/gradio_visualize_diversity.py +++ b/src/3_evals/2_synthetic_data/gradio_visualize_diversity.py @@ -4,7 +4,7 @@ uv run \ --env-file .env \ -gradio src.3_evals/2_synthetic_data/gradio_visualize_diversity.py +gradio src/3_evals/2_synthetic_data/gradio_visualize_diversity.py """ from typing import List @@ -17,8 +17,8 @@ from sklearn.decomposition import PCA from sklearn.manifold import TSNE -from ...utils import Configs, create_batches, gather_with_progress -from ...utils.langfuse.shared_client import langfuse_client +from src.utils import Configs, create_batches, gather_with_progress +from src.utils.langfuse.shared_client import langfuse_client def reduce_dimensions( @@ -79,7 +79,7 @@ async def get_projection_plot( embedding_batch_size: int = 16, ) -> Figure: """Obtain projection plot for the given dataset up to `limit` items.""" - lf_dataset_items = langfuse_client.get_dataset(dataset_name).items + lf_dataset_items = langfuse_client.get_dataset(dataset_name.strip()).items # Generate embeddings configs = Configs.from_env_var()