From 680720b09fc3413f3f6533895d4795099eccb72f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Wed, 7 Sep 2022 11:12:10 +0100
Subject: [PATCH] type(loaders): Improve custom arg typing
---
extraction/core/functions/loaders.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/extraction/core/functions/loaders.py b/extraction/core/functions/loaders.py
index 46e8ec89..cb26d47e 100644
--- a/extraction/core/functions/loaders.py
+++ b/extraction/core/functions/loaders.py
@@ -1,3 +1,4 @@
+import typing as t
from inspect import getfullargspec, getmembers, isfunction
import numpy as np
@@ -12,6 +13,7 @@ Load functions for analysing cells and their background.
Note that inspect.getmembers returns a list of function names and functions, and inspect.getfullargspec returns a function's arguments.
"""
+
def load_cellfuns_core():
"""
Load functions from the cell module and return as a dict.
@@ -24,7 +26,9 @@ def load_cellfuns_core():
}
-def load_custom_args():
+def load_custom_args() -> t.Tuple[
+ (t.Dict[str, t.Callable], t.Dict[str, t.List[str]])
+]:
"""
Load custom functions from the localisation module and return the functions and any additional arguments, other than cell_mask and trap_image, as dictionaries.
"""
--
GitLab