Browse Source

[MNT] delete 'in' after 'takes'

pull/1/head
Gao Enhao 2 years ago
parent
commit
87348bf52d
3 changed files with 4 additions and 4 deletions
  1. +1
    -1
      abl/dataset/classification_dataset.py
  2. +1
    -1
      abl/dataset/prediction_dataset.py
  3. +2
    -2
      abl/learning/basic_nn.py

+ 1
- 1
abl/dataset/classification_dataset.py View File

@@ -15,7 +15,7 @@ class ClassificationDataset(Dataset):
Y : List[int]
The target data.
transform : Callable[..., Any], optional
A function/transform that takes in an object and returns a transformed version.
A function/transform that takes an object and returns a transformed version.
Defaults to None.
"""



+ 1
- 1
abl/dataset/prediction_dataset.py View File

@@ -13,7 +13,7 @@ class PredictionDataset(Dataset):
X : List[Any]
The input data.
transform : Callable[..., Any], optional
A function/transform that takes in an object and returns a transformed version.
A function/transform that takes an object and returns a transformed version.
Defaults to None.
"""



+ 2
- 2
abl/learning/basic_nn.py View File

@@ -38,10 +38,10 @@ class BasicNN:
save_dir : Optional[str], optional
The directory in which to save the model during training, by default None.
train_transform : Callable[..., Any], optional
A function/transform that takes in an object and returns a transformed version used
A function/transform that takes an object and returns a transformed version used
in the `fit` and `train_epoch` methods, by default None.
test_transform : Callable[..., Any], optional
A function/transform that takes in an object and returns a transformed version in the
A function/transform that takes an object and returns a transformed version in the
`predict`, `predict_proba` and `score` methods, , by default None.
collate_fn : Callable[[List[T]], Any], optional
The function used to collate data, by default None.


Loading…
Cancel
Save