Update debugging.rst

ananthsub · ananthsub · commit 3a7328b17bd5 · 2021-09-24T22:49:39.000-07:00
diff --git a/docs/source/common/debugging.rst b/docs/source/common/debugging.rst
@@ -98,6 +98,7 @@ By default it only prints the top-level modules. If you want to show all submodu
 `'max_depth'` option:
 
 .. testcode::
+    from pytorch_lightning.callbacks import ModelSummary
 
     trainer = Trainer(callbacks=[ModelSummary(max_depth=-1)])
 
