float was missing from JSON type

Author: mr-c

schema_salad/avro/schema.py

@@ -58,13 +58,13 @@
 # MappingDataType = Dict[str, Union[PropType, List[PropsType]]]
 # was: Union[str, MappingDataType, List[MappingDataType]]
 JsonDataType = Any
-AtomicPropType = Union[str, int, bool, "Schema", List[str], List["Field"]]
+AtomicPropType = Union[None, str, int, float, bool, "Schema", List[str], List["Field"]]
 PropType = Union[
     AtomicPropType,
-    Dict[str, Optional[AtomicPropType]],
-    List[Dict[str, Optional[AtomicPropType]]],
+    Dict[str, AtomicPropType],
+    List[Dict[str, AtomicPropType]],
 ]
-PropsType = Dict[str, Optional[PropType]]
+PropsType = Dict[str, PropType]
 
 FIELD_RESERVED_PROPS = ("default", "name", "doc", "order", "type")
 

schema_salad/schema.py

@@ -434,8 +434,7 @@ def replace_type(
     find_embeds: bool = True,
     deepen: bool = True,
 ) -> Any:
-    """ Go through and replace types in the 'spec' mapping"""
-
+    """Go through and replace types in the 'spec' mapping."""
     if isinstance(items, MutableMapping):
         # recursively check these fields for types to replace
         if items.get("type") in ("record", "enum") and items.get("name"):

schema_salad/utils.py

@@ -98,13 +98,13 @@ def json_dump(
     fp,  # type: IO[str]
     **kwargs  # type: Any
 ):  # type: (...) -> None
-    """ Force use of unicode. """
+    """Force use of unicode."""
     json.dump(convert_to_dict(obj), fp, **kwargs)
 
 
 def json_dumps(
     obj,  # type: Any
     **kwargs  # type: Any
 ):  # type: (...) -> str
-    """ Force use of unicode. """
+    """Force use of unicode."""
     return json.dumps(convert_to_dict(obj), **kwargs)