From 367e917754ec166df53971f9d8b4b4fa89d30eb6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Michal=20Bel=C3=A1k?= Date: Fri, 1 Mar 2024 12:17:36 +0100 Subject: [PATCH] test: fix imports in multivariate analysis (#211) - Fix umap being imported when the extras is not installed, causing the test to fail - Remove unused imports --- tests/test_multivariate_analysis.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/test_multivariate_analysis.py b/tests/test_multivariate_analysis.py index 8ccdc55..5a4eb8d 100644 --- a/tests/test_multivariate_analysis.py +++ b/tests/test_multivariate_analysis.py @@ -9,7 +9,7 @@ pio.renderers.default = "json" from edvart import utils -from edvart.report_sections import multivariate_analysis, umap +from edvart.report_sections import multivariate_analysis from edvart.report_sections.code_string_formatting import code_dedent, get_code from edvart.report_sections.multivariate_analysis import ( UMAP_AVAILABLE, @@ -17,11 +17,10 @@ MultivariateAnalysisSubsection, ) from edvart.report_sections.section_base import Verbosity -from edvart.utils import ( - get_default_discrete_colorscale, - make_discrete_colorscale, - select_numeric_columns, -) +from edvart.utils import select_numeric_columns + +if UMAP_AVAILABLE: + from edvart.report_sections import umap from .execution_utils import check_section_executes from .pyarrow_utils import pyarrow_parameterize