Add additional type hints

This commit is contained in:
Pascal Kaiser 2022-04-11 10:23:11 +02:00
parent 9bc450aad8
commit 94e753cb8d
2 changed files with 2 additions and 2 deletions

View file

@ -642,7 +642,7 @@ def build_expiration(expiration_definition) -> models._ExpirationBase:
return expiration_distribution(tuple(BLO_factors))
def baseline_raw_form_data():
def baseline_raw_form_data() -> typing.Dict[str, typing.Union[str, float]]:
# Note: This isn't a special "baseline". It can be updated as required.
return {
'activity_type': 'office',

View file

@ -96,7 +96,7 @@ def interesting_times(model: models.ExposureModel, approx_n_pts=100) -> typing.L
return nice_times
def calculate_report_data(model: models.ExposureModel):
def calculate_report_data(model: models.ExposureModel) -> typing.Dict[str, typing.Any]:
times = interesting_times(model)
concentrations = [