Skip to content

Commit aef00f1

Browse files
committed
Add some missing override decorators
1 parent 07539c9 commit aef00f1

3 files changed

Lines changed: 7 additions & 0 deletions

File tree

loopy/kernel/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1345,6 +1345,7 @@ def __call__(self, *args, **kwargs):
13451345

13461346
# {{{ pickling
13471347

1348+
@override
13481349
def __getstate__(self):
13491350
result = {
13501351
fld.name: getattr(self, fld.name)

loopy/tools.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -377,6 +377,7 @@ def unpickle(self):
377377
from pickle import loads
378378
return loads(self.objstring)
379379

380+
@override
380381
def __getstate__(self):
381382
return {"objstring": self.objstring}
382383

@@ -443,6 +444,7 @@ def __len__(self):
443444
def __iter__(self) -> abc.Iterator[K]:
444445
return iter(self._map)
445446

447+
@override
446448
def __getstate__(self):
447449
return {"_map": {
448450
key: _PickledObject(val)
@@ -491,6 +493,7 @@ def __len__(self):
491493
def insert(self, key: int, value: V):
492494
self._list.insert(key, value)
493495

496+
@override
494497
def __getstate__(self):
495498
return {"_list": [_PickledObject(val) for val in self._list]}
496499

@@ -608,6 +611,7 @@ def __repr__(self):
608611
return "Optional()"
609612
return "Optional(%r)" % self._value
610613

614+
@override
611615
def __getstate__(self):
612616
if not self.has_value:
613617
return _no_value

test/test_misc.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from pickle import dumps, loads
2525

2626
import pytest
27+
from typing_extensions import override
2728

2829
import loopy as lp
2930
from loopy.version import LOOPY_USE_LANGUAGE_VERSION_2018_2 # noqa: F401
@@ -66,6 +67,7 @@ class PickleDetector:
6667
def reset(cls):
6768
cls.instance_unpickled = False
6869

70+
@override
6971
def __getstate__(self):
7072
return {"state": self.state}
7173

0 commit comments

Comments
 (0)