Skip to content

Commit 61fb081

Browse files
committed
satisfy linters
1 parent fdad917 commit 61fb081

File tree

2 files changed

+15
-9
lines changed

2 files changed

+15
-9
lines changed

notebooks/support/banzhaf.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,17 @@
1-
import numpy as np
1+
from typing import Optional
22

3+
import numpy as np
34
from numpy.typing import NDArray
45
from sklearn.datasets import load_digits
56
from sklearn.metrics import accuracy_score
67
from sklearn.model_selection import train_test_split
7-
from typing import Optional
88

99
from pydvl.utils.types import SupervisedModel
1010

1111
try:
1212
import torch
1313
from torch import nn, optim
14-
from torch.utils.data import TensorDataset, DataLoader
14+
from torch.utils.data import DataLoader, TensorDataset
1515
except ImportError as e:
1616
raise RuntimeError("PyTorch is required to run the Banzhaf MSR notebook") from e
1717

src/pydvl/value/result.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -358,13 +358,16 @@ def __getattr__(self, attr: str) -> Any:
358358
) from e
359359

360360
@overload
361-
def __getitem__(self, key: int) -> ValueItem: ...
361+
def __getitem__(self, key: int) -> ValueItem:
362+
...
362363

363364
@overload
364-
def __getitem__(self, key: slice) -> List[ValueItem]: ...
365+
def __getitem__(self, key: slice) -> List[ValueItem]:
366+
...
365367

366368
@overload
367-
def __getitem__(self, key: Iterable[int]) -> List[ValueItem]: ...
369+
def __getitem__(self, key: Iterable[int]) -> List[ValueItem]:
370+
...
368371

369372
def __getitem__(
370373
self, key: Union[slice, Iterable[int], int]
@@ -390,13 +393,16 @@ def __getitem__(
390393
raise TypeError("Indices must be integers, iterable or slices")
391394

392395
@overload
393-
def __setitem__(self, key: int, value: ValueItem) -> None: ...
396+
def __setitem__(self, key: int, value: ValueItem) -> None:
397+
...
394398

395399
@overload
396-
def __setitem__(self, key: slice, value: ValueItem) -> None: ...
400+
def __setitem__(self, key: slice, value: ValueItem) -> None:
401+
...
397402

398403
@overload
399-
def __setitem__(self, key: Iterable[int], value: ValueItem) -> None: ...
404+
def __setitem__(self, key: Iterable[int], value: ValueItem) -> None:
405+
...
400406

401407
def __setitem__(
402408
self, key: Union[slice, Iterable[int], int], value: ValueItem

0 commit comments

Comments
 (0)