Fix doctest

Lightning-Universe · Dec 6, 2021 · 945c9c7 · 945c9c7
1 parent 748abf5
commit 945c9c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: