Docs fix for atomized model (#1197)

* Docs fix
aimclub · Nov 14, 2023 · 8b16a24 · 8b16a24
1 parent 70fc033
commit 8b16a24
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/docs/source/advanced/atomized_model.rst b/docs/source/advanced/atomized_model.rst
@@ -12,10 +12,11 @@ pipeline in the AtomizedModel class and all the functionality will be saved.
 
 .. code-block:: python
 
+    from fedot.core.pipelines.node import PipelineNode
     from fedot.core.pipelines.pipeline import Pipeline
-    from fedot.core.models.atomized_model import AtomizedModel
+    from fedot.core.operations.atomized_model import AtomizedModel
 
     pipeline = Pipeline()
-    nested_pipeline = Pipeline()
-    atomized_model = AtomizedModel(nested_pipeline)
-    pipeline.add_node(atomized_model)
+    nested_pipeline = Pipeline(PipelineNode('ridge'))
+    atomized_model = PipelineNode(AtomizedModel(nested_pipeline))
+    pipeline.add_node(atomized_model)