Merge pull request #411 from RasmusOrsoe/model_class_fix

removed typo
graphnet-team · Feb 1, 2023 · 9d6ad85 · 9d6ad85
2 parents 814a702 + d17b769
commit 9d6ad85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphnet/models/model.py b/src/graphnet/models/model.py
@@ -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: