|
1 | 1 | from dataclasses import asdict, dataclass, field, is_dataclass
|
2 |
| -from typing import Callable, Literal, TypedDict |
| 2 | +from typing import Any, Callable, Literal, TypedDict |
3 | 3 |
|
4 | 4 | from typing_extensions import Unpack
|
5 | 5 |
|
@@ -381,9 +381,15 @@ def serialize(self) -> "VectorizeSettingsPartial":
|
381 | 381 | return VectorizeSettingsPartial(serialize_dataclass(self, call_self=False)) # pyright: ignore
|
382 | 382 |
|
383 | 383 | @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) |
388 | 394 |
|
389 | 395 | return VectorizeSettings(**d)
|
0 commit comments