|
15 | 15 | from rocketpy.plots.monte_carlo_plots import _MonteCarloPlots
|
16 | 16 | from rocketpy.prints.monte_carlo_prints import _MonteCarloPrints
|
17 | 17 | from rocketpy.simulation.flight import Flight
|
18 |
| -from rocketpy.simulation.sim_config.flight2serializer import flightv1_serializer |
| 18 | +from rocketpy.simulation.sim_config.flight2serializer import \ |
| 19 | + flightv1_serializer |
19 | 20 | from rocketpy.simulation.sim_config.serializer import function_serializer
|
20 |
| -from rocketpy.stochastic import ( |
21 |
| - StochasticEnvironment, |
22 |
| - StochasticFlight, |
23 |
| - StochasticRocket, |
24 |
| -) |
25 |
| -from rocketpy.tools import ( |
26 |
| - generate_monte_carlo_ellipses, |
27 |
| - generate_monte_carlo_ellipses_coordinates, |
28 |
| -) |
| 21 | +from rocketpy.stochastic import (StochasticEnvironment, StochasticFlight, |
| 22 | + StochasticRocket) |
| 23 | +from rocketpy.tools import (generate_monte_carlo_ellipses, |
| 24 | + generate_monte_carlo_ellipses_coordinates) |
29 | 25 |
|
30 | 26 | # TODO: Let Functions and Flights be json serializable
|
31 | 27 | # TODO: Create evolution plots to analyze convergence
|
@@ -184,10 +180,8 @@ def simulate(
|
184 | 180 | None
|
185 | 181 | """
|
186 | 182 | if append and light_mode is False:
|
187 |
| - raise ValueError( |
188 |
| - "Append mode is not available when light_mode is False." |
189 |
| - ) |
190 |
| - |
| 183 | + raise ValueError("Append mode is not available when light_mode is False.") |
| 184 | + |
191 | 185 | # initialize counters
|
192 | 186 | self.number_of_simulations = number_of_simulations
|
193 | 187 | self.iteration_count = self.num_of_loaded_sims if append else 0
|
@@ -237,7 +231,9 @@ def _run_in_serial(self, append, light_mode):
|
237 | 231 | self.__close_files(input_file, output_file, error_file)
|
238 | 232 | raise error
|
239 | 233 |
|
240 |
| - self.__finalize_simulation(input_file, output_file, error_file, light_mode=light_mode) |
| 234 | + self.__finalize_simulation( |
| 235 | + input_file, output_file, error_file, light_mode=light_mode |
| 236 | + ) |
241 | 237 |
|
242 | 238 | def _run_in_parallel(self, append, light_mode, n_workers=None):
|
243 | 239 | """Runs the monte carlo simulation in parallel."""
|
|
0 commit comments