From 18b2ebe3ff4569c3dcc646488727d9547844545f Mon Sep 17 00:00:00 2001 From: Pedro Bressan Date: Wed, 16 Aug 2023 21:45:36 -0300 Subject: [PATCH] FIX: optional imports for EnvironmentAnalysis prevent default install. --- rocketpy/EnvironmentAnalysis.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/rocketpy/EnvironmentAnalysis.py b/rocketpy/EnvironmentAnalysis.py index c661ebca7..7799a9dd8 100644 --- a/rocketpy/EnvironmentAnalysis.py +++ b/rocketpy/EnvironmentAnalysis.py @@ -18,18 +18,6 @@ from rocketpy.Environment import Environment from rocketpy.Function import Function from rocketpy.units import convert_units - -try: - import ipywidgets as widgets - import jsonpickle - from timezonefinder import TimezoneFinder - from windrose import WindroseAxes -except ImportError as error: - raise ImportError( - f"The following error was encountered while importing dependencies: '{error}'. " - "Please note that the EnvironmentAnalysis requires additional dependencies, " - "which can be installed by running 'pip install rocketpy[env_analysis]'." - ) from .plots.environment_analysis_plots import _EnvironmentAnalysisPlots from .prints.environment_analysis_prints import _EnvironmentAnalysisPrints from .tools import ( @@ -180,7 +168,19 @@ def __init__( ------- None """ - + # Import optional modules + try: + import ipywidgets as widgets + import jsonpickle + from timezonefinder import TimezoneFinder + from windrose import WindroseAxes + except ImportError as error: + raise ImportError( + "The following error was encountered while importing dependencies: " + f"{error}. Please note that the EnvironmentAnalysis requires additional " + "dependencies, which can be installed by running 'pip install " + "rocketpy[env_analysis]'." + ) # Save inputs self.start_date = start_date self.end_date = end_date