Skip to content

Commit

Permalink
Support polars in hvplot.plot method (#1219)
Browse files Browse the repository at this point in the history
* support polars in hvplot.plot method

* Update hvplot/util.py

Co-authored-by: Simon Høxbro Hansen <simon.hansen@me.com>

---------

Co-authored-by: Simon Høxbro Hansen <simon.hansen@me.com>
  • Loading branch information
MarcoGorelli and hoxbro committed Dec 20, 2023
1 parent e37715b commit ab61de5
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 1 deletion.
5 changes: 4 additions & 1 deletion hvplot/plotting/__init__.py
@@ -1,5 +1,5 @@
import holoviews as hv
from ..util import with_hv_extension
from ..util import with_hv_extension, is_polars

from .core import hvPlot, hvPlotTabular # noqa

Expand Down Expand Up @@ -30,6 +30,9 @@ def plot(data, kind, **kwargs):
if v is not None:
no_none_kwargs[k] = v

if is_polars(data):
from hvplot.polars import hvPlotTabularPolars
return hvPlotTabularPolars(data)(kind=kind, **no_none_kwargs)
return hvPlotTabular(data)(kind=kind, **no_none_kwargs)


Expand Down
6 changes: 6 additions & 0 deletions hvplot/util.py
Expand Up @@ -348,6 +348,12 @@ def is_dask(data):
import dask.dataframe as dd
return isinstance(data, (dd.DataFrame, dd.Series))

def is_polars(data):
if not check_library(data, 'polars'):
return False
import polars as pl
return isinstance(data, (pl.DataFrame, pl.Series, pl.LazyFrame))

def is_intake(data):
if "intake" not in sys.modules:
return False
Expand Down

0 comments on commit ab61de5

Please sign in to comment.