fix(rich): Move import into scoped block (#193)

automl · Dec 6, 2023 · adf8d0b · adf8d0b
1 parent 8862429
commit adf8d0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/amltk/pipeline/node.py b/src/amltk/pipeline/node.py
@@ -57,7 +57,6 @@
 from typing_extensions import override
 
 from more_itertools import first_true
-from rich.text import Text
 from sklearn.pipeline import Pipeline as SklearnPipeline
 
 from amltk._functional import classname, mapping_select, prefix_keys
@@ -714,6 +713,7 @@ def display(self, *, full: bool = False) -> RenderableType:
     def __rich__(self) -> Panel:
         from rich.console import Group as RichGroup
         from rich.panel import Panel
+        from rich.text import Text
 
         clr = self.RICH_OPTIONS.panel_color
         title = Text.assemble(