diff --git a/edvart/report.py b/edvart/report.py index fead723..1624ebe 100755 --- a/edvart/report.py +++ b/edvart/report.py @@ -150,12 +150,14 @@ def _generate_notebook( # Add imports cell imports_set = { - "import pandas as pd", "import os", "from typing import Any, Callable, Dict, List, Optional, Tuple, Union", + "", + "import pandas as pd", + "import plotly.io as pio", "import plotly.offline as py", + "", "py.init_notebook_mode()", - "import plotly.io as pio", "pio.renderers.default = 'plotly_mimetype+notebook'", } if extra_imports is not None: