We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a315d2 commit 652db10Copy full SHA for 652db10
autoemulate/emulators/lightgbm.py
@@ -157,7 +157,12 @@ def is_multioutput() -> bool:
157
"""LightGBM does not support multi-output."""
158
return False
159
160
- def _fit(self, x: TensorLike, y: TensorLike): # type: ignore since this is valid subclass of types
+ def _fit(
161
+ self,
162
+ x: TensorLike,
163
+ y: TensorLike,
164
+ validation_data: tuple[TensorLike, TensorLike] | None = None, # noqa: ARG002
165
+ ) -> None:
166
x_np, y_np = self._convert_to_numpy(x, y)
167
self.n_features_in_ = x_np.shape[1]
168
self.model_.fit(x_np, y_np)
0 commit comments