Skip to content

Commit e7dccf5

Browse files
authored
📝 Type-hint and docstring fix (#44)
1 parent 8566ac5 commit e7dccf5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/ml_workshop/_penguins.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Penguins dataset."""
2+
23
from typing import Optional, List, Dict, Tuple, Any
34

45
from torch.utils.data import Dataset
@@ -17,9 +18,9 @@ class PenguinDataset(Dataset):
1718
1819
Parameters
1920
----------
20-
input_keys : Sequence[str]
21+
input_keys : List[str]
2122
The column titles to use in the input feature vectors.
22-
target_keys : Sequnce[str]
23+
target_keys : List[str]
2324
The column titles to use in the target feature vectors.
2425
train : bool
2526
If ``True``, this object will serve as the training set, and if
@@ -39,7 +40,7 @@ class PenguinDataset(Dataset):
3940
def __init__(
4041
self,
4142
input_keys: List[str],
42-
target_keys: str,
43+
target_keys: List[str],
4344
train: bool,
4445
x_tfms: Optional[Compose] = None,
4546
y_tfms: Optional[Compose] = None,

0 commit comments

Comments
 (0)