Skip to content

Commit

Permalink
Fixup type annotation style
Browse files Browse the repository at this point in the history
  • Loading branch information
mmore500 committed Dec 26, 2023
1 parent 35c2fcb commit f20b163
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions src/outset/patched/_regplot.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,18 +4,17 @@
import pandas as pd
from matplotlib import pyplot as plt
import seaborn as sns
from typing import Optional, Any, Dict


def regplot(
data: pd.DataFrame,
*,
x: str,
y: str,
hue: Optional[str] = None,
hue_order: Optional[Any] = None,
ax: Optional[plt.Axes] = None,
**kwargs: Dict[str, Any],
hue: typing.Optional[str] = None,
hue_order: typing.Optional[typing.Any] = None,
ax: typing.Optional[plt.Axes] = None,
**kwargs: dict,
) -> plt.Axes:
"""Plot regressions with seaborn's regplot on a pandas DataFrame.
Expand All @@ -33,7 +32,7 @@ def regplot(
The order to plot the `hue` levels, if `hue` is not None.
ax : Optional[plt.Axes], default None
The matplotlib Axes object to draw the plot onto, if provided.
**kwargs
**kwargs : dict
Additional keyword arguments forward to seaborn's regplot.
Returns
Expand Down

0 comments on commit f20b163

Please sign in to comment.