
    eCi                    x    S SK Jr  S SKJr  S SKrS SKrS SKJ	r	  \ " S S\	5      5       r
\ " S S\	5      5       rg)	    )annotations)	dataclassN)Statc                  B    \ rS rSr% SrSrS\S'   SrS\S'   S rS	 r	S
r
g)PolyFit
   zM
Fit a polynomial of the given order and resample data onto predicted curve.
   intorderd   gridsizec                ~   US   nUS   nUR                  5       U R                  ::  a  / =pEOu[        R                  " X#U R                  5      n[        R                  " UR                  5       UR                  5       U R                  5      n[        R                  " Xd5      n[        R                  " [        XES95      $ )Nxy)r   r   )nuniquer   nppolyfitlinspaceminmaxr   polyvalpd	DataFramedict)selfdatar   r   xxyyps          S/home/james-whalen/.local/lib/python3.13/site-packages/seaborn/_stats/regression.py_fit_predictPolyFit._fit_predict   s    II99;$**$LB

1,AQUUWaeegt}}=BA"B||D2,--    c                X    UR                  UR                  SS/S9U R                  5      $ )Nr   r   )subset)applydropnar!   )r   r   groupbyorientscaless        r    __call__PolyFit.__call__'   s-     U4;;sCj;143D3DE	
r#    N)__name__
__module____qualname____firstlineno____doc__r   __annotations__r   r!   r+   __static_attributes__r-   r#   r    r   r   
   s&     E3NHc."
r#   r   c                      \ rS rSrSrg)OLSFit/   r-   N)r.   r/   r0   r1   r4   r-   r#   r    r6   r6   /   s     r#   r6   )
__future__r   dataclassesr   numpyr   pandasr   seaborn._stats.baser   r   r6   r-   r#   r    <module>r=      sK    " !   $ !
d !
 !
H T  r#   