Fix import in doctest example (#14067)

Lightning-AI · Aug 9, 2022 · d29a552 · d29a552
1 parent 0cfc53d
commit d29a552
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/lightning_app/structures/dict.py b/src/lightning_app/structures/dict.py
@@ -22,7 +22,7 @@ def __init__(self, **kwargs: T):
         .. doctest::
 
             >>> from lightning_app import LightningFlow, LightningWork
-            >>> from lightning_app.core import Dict
+            >>> from lightning_app.structures import Dict
             >>> class CounterWork(LightningWork):
             ...     def __init__(self):
             ...         super().__init__()

diff --git a/src/lightning_app/structures/list.py b/src/lightning_app/structures/list.py
@@ -24,7 +24,7 @@ def __init__(self, *items: T):
         .. doctest::
 
             >>> from lightning_app import LightningFlow, LightningWork
-            >>> from lightning_app.core import List
+            >>> from lightning_app.structures import List
             >>> class CounterWork(LightningWork):
             ...     def __init__(self):
             ...         super().__init__()