From 945c9c7910bc4dc5e4f1548da0099667807077b4 Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Mon, 6 Dec 2021 19:41:44 +0000 Subject: [PATCH] Fix doctest --- docs/source/quickstart.rst | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/source/quickstart.rst b/docs/source/quickstart.rst index b8a120dae3..a729684630 100644 --- a/docs/source/quickstart.rst +++ b/docs/source/quickstart.rst @@ -103,6 +103,7 @@ Here's an example of inference: "This guy has done a great job with this movie!", ] ) + predictions = trainer.predict(model, datamodule=datamodule) print(predictions) We get the following output: