Skip to content

fix(223): Replace deprecated _get_shared_lib_location with register_plugin_function #245

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

fix(223): Replace deprecated _get_shared_lib_location with register_plugin_function #245

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 16 additions & 14 deletions functime/feature_extractors.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,13 @@

import logging
import math
from pathlib import Path
from typing import List, Mapping, Optional, Sequence, Union

import numpy as np
import polars as pl
from polars.plugins import register_plugin_function
from polars.type_aliases import ClosedInterval
from polars.utils.udfs import _get_shared_lib_location

# from numpy.linalg import lstsq
from scipy.linalg import lstsq
Expand All @@ -34,8 +35,6 @@

# from polars.type_aliases import IntoExpr

lib = _get_shared_lib_location(__file__)


def absolute_energy(x: TIME_SERIES_T) -> FLOAT_INT_EXPR:
"""
Expand Down Expand Up @@ -2255,9 +2254,10 @@ def lempel_ziv_complexity(
https://github.com/Naereen/Lempel-Ziv_Complexity/tree/master
https://en.wikipedia.org/wiki/Lempel%E2%80%93Ziv_complexity
"""
out = (self._expr > threshold).register_plugin(
lib=lib,
symbol="pl_lempel_ziv_complexity",
out = register_plugin_function(
plugin_path=Path(__file__).parent,
function_name="pl_lempel_ziv_complexity",
args=self._expr > threshold,
is_elementwise=False,
returns_scalar=True,
)
Expand Down Expand Up @@ -2766,16 +2766,17 @@ def cusum(
-------
An expression of the output
"""
return self._expr.register_plugin(
lib=lib,
symbol="cusum",
return register_plugin_function(
plugin_path=Path(__file__).parent,
function_name="cusum",
args=self._expr,
kwargs={
"threshold": threshold,
"drift": drift,
"warmup_period": warmup_period,
},
is_elementwise=False,
cast_to_supertypes=True,
cast_to_supertype=True,
)

def frac_diff(
Expand Down Expand Up @@ -2815,14 +2816,15 @@ def frac_diff(
if min_weight is None and window_size is None:
raise ValueError("Either min_weight or window_size must be specified.")

return self._expr.register_plugin(
lib=lib,
symbol="frac_diff",
return register_plugin_function(
plugin_path=Path(__file__).parent,
function_name="frac_diff",
args=self._expr,
kwargs={
"d": d,
"min_weight": min_weight,
"window_size": window_size,
},
is_elementwise=False,
cast_to_supertypes=True,
cast_to_supertype=True,
)