diff --git a/plotly/figure_factory/_streamline.py b/plotly/figure_factory/_streamline.py index ddc14778c43..a6773420f48 100644 --- a/plotly/figure_factory/_streamline.py +++ b/plotly/figure_factory/_streamline.py @@ -354,7 +354,11 @@ def get_streamline_arrows(self): dif_x = arrow_end_x - arrow_start_x dif_y = arrow_end_y - arrow_start_y + orig_err = np.geterr() + np.seterr(divide='ignore', invalid='ignore') streamline_ang = np.arctan(dif_y / dif_x) + np.seterr(**orig_err) + ang1 = streamline_ang + (self.angle) ang2 = streamline_ang - (self.angle)