Fixes {#19}

NKI-AI · Dec 8, 2023 · b54ec38 · b54ec38
1 parent 041c515
commit b54ec38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ahcore/entrypoints.py b/ahcore/entrypoints.py
@@ -268,7 +268,7 @@ def inference(config: DictConfig) -> None:
 
     # Inference
     logger.info("Starting inference...")
-    trainer.predict(model=model, datamodule=datamodule)
+    trainer.predict(model=model, datamodule=datamodule, return_predictions=False)
 
     # Make sure everything closed properly
     logger.info("Finalizing...")