adds more missing typing

Author: tomjemmett

src/nhp/model/data/local.py

@@ -112,7 +112,7 @@ def get_inequalities(self) -> pd.DataFrame:
         """
         return self._get_parquet("inequalities")
 
-    def _get_parquet(self, file) -> pd.DataFrame:
+    def _get_parquet(self, file: str) -> pd.DataFrame:
         """Load specific parquet file using Pandas.
 
         Args:

src/nhp/model/results.py

@@ -76,7 +76,7 @@ def _combine_model_results(
     }
 
 
-def _combine_step_counts(results: list):
+def _combine_step_counts(results: list) -> pd.DataFrame:
     """Combine the step counts of the monte carlo runs.
 
     Takes as input a list of lists, where the outer list contains an item for inpatients,

src/nhp/model/run.py

@@ -47,7 +47,7 @@ def _run_model(
     data: Callable[[int, str], Data],
     hsa: Any,
     run_params: dict,
-    progress_callback,
+    progress_callback: Callable[[Any], None],
     save_full_model_results: bool,
 ) -> list[ModelRunResult]:
     """Run the model iterations.