Improved type annotations

labthings · Jan 5, 2021 · 6e8b47e · 6e8b47e
1 parent 37919b1
commit 6e8b47e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/labthings/marshalling/marshalling.py b/src/labthings/marshalling/marshalling.py
@@ -1,6 +1,6 @@
 from collections.abc import Mapping
 from functools import wraps
-from typing import Callable, Dict, Tuple, Union
+from typing import Callable, Dict, Tuple, Union, Optional
 
 from marshmallow import Schema as _Schema
 from werkzeug.wrappers import Response as ResponseBase
@@ -10,7 +10,9 @@
 from ..utilities import unpack
 
 
-def schema_to_converter(schema: Union[Schema, Field, Dict[str, Union[Field, type]]]):
+def schema_to_converter(
+    schema: Union[Schema, Field, Dict[str, Union[Field, type]]]
+) -> Optional[Callable]:
     """Convert a schema into a converter function,
     which takes a value as an argument and returns
     marshalled data