From ae0d9efef2c9849624c7619463c8b613cd3d3e31 Mon Sep 17 00:00:00 2001 From: Beatriz Saldana Date: Mon, 30 Sep 2024 16:28:14 -0700 Subject: [PATCH] Added output type hint to apply_custom_transformations() --- src/agoradatatools/process.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/agoradatatools/process.py b/src/agoradatatools/process.py index d25d6a0..fdc26e8 100644 --- a/src/agoradatatools/process.py +++ b/src/agoradatatools/process.py @@ -14,11 +14,15 @@ from agoradatatools.reporter import ADTGXReporter, DatasetReport from agoradatatools.constants import Platform +from typing import Union + logger = logging.getLogger(__name__) # TODO refactor to avoid so many if's - maybe some sort of mapping to callables -def apply_custom_transformations(datasets: dict, dataset_name: str, dataset_obj: dict): +def apply_custom_transformations( + datasets: dict, dataset_name: str, dataset_obj: dict +) -> Union[DataFrame, list, dict]: if not isinstance(datasets, dict) or not isinstance(dataset_name, str): return None if dataset_name == "biodomain_info":