forward kwargs

basf · Apr 23, 2024 · 0840a96 · 0840a96
1 parent d041a2f
commit 0840a96
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/molpipeline/estimators/chemprop/models.py b/molpipeline/estimators/chemprop/models.py
@@ -183,7 +183,7 @@ def __init__(
         lightning_trainer: pl.Trainer | None = None,
         batch_size: int = 64,
         n_jobs: int = 1,
-        **kwargs: Any,  # pylint: disable=unused-argument
+        **kwargs: Any,
     ) -> None:
         """Initialize the chemprop classifier model.
 
@@ -211,6 +211,7 @@ def __init__(
             lightning_trainer=lightning_trainer,
             batch_size=batch_size,
             n_jobs=n_jobs,
+            **kwargs,
         )
         if not self._is_binary_classifier():
             raise ValueError("ChempropClassifier should be a binary classifier.")
@@ -243,7 +244,7 @@ def __init__(
         lightning_trainer: pl.Trainer | None = None,
         batch_size: int = 64,
         n_jobs: int = 1,
-        **kwargs: Any,  # pylint: disable=unused-argument
+        **kwargs: Any,
     ) -> None:
         """Initialize the chemprop regressor model.
 
@@ -271,4 +272,5 @@ def __init__(
             lightning_trainer=lightning_trainer,
             batch_size=batch_size,
             n_jobs=n_jobs,
+            **kwargs,
         )