diff --git a/abl/dataset/classification_dataset.py b/abl/dataset/classification_dataset.py
index 091c3b5..2efb8f6 100644
--- a/abl/dataset/classification_dataset.py
+++ b/abl/dataset/classification_dataset.py
@@ -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.
     """
 
diff --git a/abl/dataset/prediction_dataset.py b/abl/dataset/prediction_dataset.py
index 7674993..6776988 100644
--- a/abl/dataset/prediction_dataset.py
+++ b/abl/dataset/prediction_dataset.py
@@ -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.
     """
 
diff --git a/abl/learning/basic_nn.py b/abl/learning/basic_nn.py
index b2767b7..40effee 100644
--- a/abl/learning/basic_nn.py
+++ b/abl/learning/basic_nn.py
@@ -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.