Skip to content

Commit d621adc

Browse files
committed
consistent skipkeys type annotation
1 parent 3816aeb commit d621adc

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/superneuromat/neuromorphicmodel.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2133,7 +2133,8 @@ def clear_memos(self):
21332133
del self.memoized
21342134
self.memoized = {}
21352135

2136-
def _to_json_dict(self, array_representation="json-native", skipkeys=None,
2136+
def _to_json_dict(self, array_representation="json-native",
2137+
skipkeys: list[str] | tuple[str] | set[str] | None = None,
21372138
net_name=None, extra=None):
21382139
"""Exports the SNN to a dictionary.
21392140
@@ -2251,7 +2252,8 @@ def get_dtype(o):
22512252
d["networks"].append(networkd)
22522253
return d
22532254

2254-
def to_json(self, array_representation="json-native", skipkeys: list[str] | None = None,
2255+
def to_json(self, array_representation="json-native",
2256+
skipkeys: list[str] | tuple[str] | set[str] | None = None,
22552257
net_name: str | None = None, extra: dict | None = None, indent=2, **kwargs):
22562258
"""Exports the SNN to a JSON string.
22572259
@@ -2291,7 +2293,8 @@ def to_json(self, array_representation="json-native", skipkeys: list[str] | None
22912293
return json.dumps(d, indent=indent, **kwargs)
22922294

22932295
def saveas_json(self, fp,
2294-
array_representation="json-native", skipkeys: list[str] | None = None,
2296+
array_representation="json-native",
2297+
skipkeys: list[str] | tuple[str] | set[str] | None = None,
22952298
net_name: str | None = None, extra: dict | None = None, indent=2, **kwargs):
22962299
"""Exports the SNN to a JSON file.
22972300

0 commit comments

Comments
 (0)