Skip to content

Commit c2958df

Browse files
author
Neumann, Jan
committed
Bugfix in settings deserialize
1 parent 1445116 commit c2958df

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

lib/vectorize/settings.py

+11-5
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from dataclasses import asdict, dataclass, field, is_dataclass
2-
from typing import Callable, Literal, TypedDict
2+
from typing import Any, Callable, Literal, TypedDict
33

44
from typing_extensions import Unpack
55

@@ -381,9 +381,15 @@ def serialize(self) -> "VectorizeSettingsPartial":
381381
return VectorizeSettingsPartial(serialize_dataclass(self, call_self=False)) # pyright: ignore
382382

383383
@staticmethod
384-
def deserialize(d: VectorizeSettingsPartial):
385-
d["linears_symmetries"] = LinearsSymmetriesSettings(**d["linears_symmetries"]) # pyright: ignore
386-
d["optimize_tail_refs"] = OptimizeTailRefsSettings(**d["optimize_tail_refs"]) # pyright: ignore
387-
d["optimize_single_use_gathers"] = OptimizeSingleUseGathersSettings(**d["optimize_single_use_gathers"]) # pyright: ignore
384+
def deserialize(d: dict[str, Any]):
385+
linears_symmetries = d.get("linears_symmetries", None)
386+
optimize_tail_refs = d.get("optimize_tail_refs", None)
387+
optimize_single_use_gathers = d.get("optimize_single_use_gathers", None)
388+
if linears_symmetries:
389+
d["linears_symmetries"] = LinearsSymmetriesSettings(**linears_symmetries)
390+
if optimize_tail_refs:
391+
d["optimize_tail_refs"] = OptimizeTailRefsSettings(**optimize_tail_refs)
392+
if optimize_single_use_gathers:
393+
d["optimize_single_use_gathers"] = OptimizeSingleUseGathersSettings(**optimize_single_use_gathers)
388394

389395
return VectorizeSettings(**d)

0 commit comments

Comments
 (0)