diff --git a/caimira/apps/calculator/report_generator.py b/caimira/apps/calculator/report_generator.py index 432cf286..885c2e8d 100644 --- a/caimira/apps/calculator/report_generator.py +++ b/caimira/apps/calculator/report_generator.py @@ -15,7 +15,7 @@ import matplotlib.pyplot as plt from caimira import models from caimira.apps.calculator import markdown_tools from ... import monte_carlo as mc -from .model_generator import FormData, _DEFAULT_MC_SAMPLE_SIZE +from .model_generator import FormData, DEFAULT_MC_SAMPLE_SIZE from ... import dataclass_utils @@ -362,7 +362,7 @@ def manufacture_alternative_scenarios(form: FormData) -> typing.Dict[str, mc.Exp def scenario_statistics(mc_model: mc.ExposureModel, sample_times: typing.List[float], compute_prob_exposure: bool): - model = mc_model.build_model(size=_DEFAULT_MC_SAMPLE_SIZE) + model = mc_model.build_model(size=DEFAULT_MC_SAMPLE_SIZE) if (compute_prob_exposure): # It means we have data to calculate the total_probability_rule prob_probabilistic_exposure = model.total_probability_rule() diff --git a/caimira/tests/apps/calculator/test_model_generator.py b/caimira/tests/apps/calculator/test_model_generator.py index 787e7a8e..099a00d2 100644 --- a/caimira/tests/apps/calculator/test_model_generator.py +++ b/caimira/tests/apps/calculator/test_model_generator.py @@ -540,7 +540,7 @@ def test_default_types(): # Handle typing.NewType definitions. field_type = field_type.__supertype__ - if value is model_generator._NO_DEFAULT: + if value is model_generator.NO_DEFAULT: continue if field in model_generator._CAST_RULES_FORM_ARG_TO_NATIVE: