Commit ea83ea8b authored by Duggan, John's avatar Duggan, John
Browse files

Fix plotly rendering issue

parent 19cdfb52

code/episode_7/src/nova_tutorial/models/plotly.py

+8 −2
Original line number Diff line number Diff line
@@ -24,9 +24,15 @@ class PlotlyConfig(BaseModel): 
    def get_figure(self) -> go.Figure:

        match self.plot_type:

            case "heatmap":

                plot_data = go.Heatmap(x=IRIS_DATA[self.x_axis], y=IRIS_DATA[self.y_axis], z=IRIS_DATA[self.z_axis])

                plot_data = go.Heatmap(

                    x=IRIS_DATA[self.x_axis].tolist(),

                    y=IRIS_DATA[self.y_axis].tolist(),

                    z=IRIS_DATA[self.z_axis].tolist(),

                )

            case "scatter":

                plot_data = go.Scatter(x=IRIS_DATA[self.x_axis], y=IRIS_DATA[self.y_axis], mode="markers")

                plot_data = go.Scatter(

                    x=IRIS_DATA[self.x_axis].tolist(), y=IRIS_DATA[self.y_axis].tolist(), mode="markers"

                )

            case _:

                raise ValueError(f"Invalid plot type: {self.plot_type}")

episodes/07-Advanced-Visualizations.md

+10 −2
Original line number Diff line number Diff line
@@ -155,9 +155,17 @@ class PlotlyConfig(BaseModel): 
    def get_figure(self) -> go.Figure:

        match self.plot_type:

            case "heatmap":

                plot_data = go.Heatmap(x=IRIS_DATA[self.x_axis], y=IRIS_DATA[self.y_axis], z=IRIS_DATA[self.z_axis])

                plot_data = go.Heatmap(

                    x=IRIS_DATA[self.x_axis].tolist(),

                    y=IRIS_DATA[self.y_axis].tolist(),

                    z=IRIS_DATA[self.z_axis].tolist()

                )

            case "scatter":

                plot_data = go.Scatter(x=IRIS_DATA[self.x_axis], y=IRIS_DATA[self.y_axis], mode="markers")

                plot_data = go.Scatter(

                    x=IRIS_DATA[self.x_axis].tolist(),

                    y=IRIS_DATA[self.y_axis].tolist(),

                    mode="markers"

                )

            case _:

                raise ValueError(f"Invalid plot type: {self.plot_type}")