dataclass-extensions 0.2.1__tar.gz → 0.2.3__tar.gz

{dataclass_extensions-0.2.1 → dataclass_extensions-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataclass-extensions
-Version: 0.2.1
+Version: 0.2.3
 Summary: Additional functionality for Python dataclasses
 Author-email: Pete Walsh <epwalsh10@gmail.com>
 License:                                  Apache License

{dataclass_extensions-0.2.1 → dataclass_extensions-0.2.3}/pyproject.toml

@@ -91,13 +91,9 @@ exclude = [
     ".tox",
     "__pypackages__",
     "_build",
-    "buck-out",
     "build",
     "dist",
-    "node_modules",
-    "doc",
-    "pretrain_data",
-    "inference",
+    "src/test/_type_alias.py",
 ]
 
 [tool.ruff.lint.per-file-ignores]

{dataclass_extensions-0.2.1 → dataclass_extensions-0.2.3}/src/dataclass_extensions/decode.py

@@ -26,7 +26,11 @@ class Decoder:
         Decode a dataset from a JSON-safe dictionary. The inverse of :func:`encode()`.
         """
         type_hints = typing.get_type_hints(config_class)
-        kwargs = {k: _coerce(v, type_hints[k], self.custom_handlers, k) for k, v in data.items()}
+        kwargs: dict[str, Any] = {}
+        for k, v in data.items():
+            if k not in type_hints:
+                raise AttributeError(f"class '{config_class.__qualname__}' has no attribute '{k}'")
+            kwargs[k] = _coerce(v, type_hints[k], self.custom_handlers, k)
         return config_class(**kwargs)
 
 
@@ -96,6 +100,9 @@ def _coerce(
             except TypeError:
                 pass
 
+        if allowed_type is float and _safe_isinstance(value, (float, int)):
+            return float(value)
+
         origin = getattr(allowed_type, "__origin__", None)
         args = getattr(allowed_type, "__args__", None)
         if (origin is list or origin is collections.abc.MutableSequence) and _safe_isinstance(
@@ -168,7 +175,14 @@ def _coerce(
                 type_name = value.get("type", allowed_type._default_type)
                 if type_name is not None:
                     allowed_type = allowed_type.get_registered_class(type_name)
-            type_hints = typing.get_type_hints(allowed_type)
+
+            try:
+                type_hints = typing.get_type_hints(allowed_type)
+            except NameError as e:
+                raise NameError(
+                    f"{str(e)}. If you're using 'from __future__ import annotations' you may need to import this type."
+                ) from e
+
             kwargs = {}
             for k, v in value.items():
                 try:

{dataclass_extensions-0.2.1 → dataclass_extensions-0.2.3}/src/dataclass_extensions/encode.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import dataclasses
 import pathlib
 from datetime import datetime
+from enum import Enum
 from typing import Any, Callable, ClassVar, Generator, Type, TypeVar
 
 from .registrable import Registrable
@@ -70,6 +71,8 @@ class Encoder:
                 return d.timestamp()
             elif isinstance(d, pathlib.Path):
                 return str(d)
+            elif isinstance(d, Enum):
+                return d.value
             elif d is None or isinstance(d, (float, int, bool, str)):
                 return d
 

dataclass_extensions-0.2.3/src/dataclass_extensions/version.py

@@ -0,0 +1 @@
+VERSION = "0.2.3"

{dataclass_extensions-0.2.1 → dataclass_extensions-0.2.3}/src/dataclass_extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataclass-extensions
-Version: 0.2.1
+Version: 0.2.3
 Summary: Additional functionality for Python dataclasses
 Author-email: Pete Walsh <epwalsh10@gmail.com>
 License:                                  Apache License

dataclass_extensions-0.2.1/src/dataclass_extensions/version.py

@@ -1 +0,0 @@
-VERSION = "0.2.1"