Skip to content

Commit d70f2d4

Browse files
authored
Fix Mypy 1.11 false positive (#597)
1 parent 6a433a6 commit d70f2d4

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

libs/knowledge-store/ragstack_knowledge_store/graph_store.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import json
22
import re
33
import secrets
4-
from dataclasses import dataclass, field
4+
from dataclasses import asdict, dataclass, field, is_dataclass
55
from enum import Enum
66
from typing import (
77
Any,
@@ -57,12 +57,11 @@ def _serialize_metadata(md: Dict[str, Any]) -> str:
5757

5858

5959
def _serialize_links(links: Set[Link]) -> str:
60-
import dataclasses
61-
6260
class SetAndLinkEncoder(json.JSONEncoder):
6361
def default(self, obj: Any) -> Any:
64-
if dataclasses.is_dataclass(obj):
65-
return dataclasses.asdict(obj)
62+
if is_dataclass(obj):
63+
# TODO: Mypy false positive: https://github.com/python/mypy/issues/17550
64+
return asdict(obj) # type: ignore[call-overload]
6665

6766
try:
6867
iterable = iter(obj)

0 commit comments

Comments
 (0)