Skip to content

Commit dc4e77c

Browse files
authored
Remove method as in base class
1 parent 8d82184 commit dc4e77c

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

autoemulate/experimental/emulators/nn/mlp_gaussian.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -73,11 +73,6 @@ def __init__(
7373
self.scheduler_setup(scheduler_kwargs)
7474
self.to(device)
7575

76-
def _predict(self, x, with_grad=False):
77-
"""Predict using the MLP model."""
78-
with torch.set_grad_enabled(with_grad):
79-
self.nn.eval()
80-
return self(x)
8176

8277
def forward(self, x):
8378
"""Forward pass for the Gaussian MLP."""

0 commit comments

Comments
 (0)