diff --git a/simba/plotting/gantt_plotly.py b/simba/plotting/gantt_plotly.py index 80c738d7d..f75f4e653 100644 --- a/simba/plotting/gantt_plotly.py +++ b/simba/plotting/gantt_plotly.py @@ -1,13 +1,17 @@ -import pandas as pd -import numpy as np -import plotly.express as px -import PIL import io from typing import Optional, Tuple, Union + +import numpy as np +import pandas as pd +import PIL +import plotly.express as px + +from simba.utils.checks import (check_float, check_instance, check_int, + check_valid_dataframe) from simba.utils.data import create_color_palettes, detect_bouts -from simba.utils.checks import check_float, check_int, check_instance, check_valid_dataframe -from simba.utils.read_write import seconds_to_timestamp, read_df from simba.utils.errors import InvalidInputError +from simba.utils.read_write import read_df, seconds_to_timestamp + def gantt_plotly(bouts_df: pd.DataFrame, img_size: Optional[Tuple[int, int]] = (640, 480),