Skip to content

Commit

Permalink
Merge pull request graphnet-team#411 from RasmusOrsoe/model_class_fix
Browse files Browse the repository at this point in the history
removed typo
  • Loading branch information
RasmusOrsoe authored Feb 1, 2023
2 parents a924e5b + 09bf2c0 commit 33ac7ea
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/graphnet/models/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,7 @@ def predict(
self.train(mode=False)

if not hasattr(self, "_inference_trainer"):
self._construct_trainer(
self._construct_trainers(
gpus=gpus,
)
elif gpus is not None:
Expand Down

0 comments on commit 33ac7ea

Please sign in to comment.