Skip to content

Commit f330473

Browse files
committed
Fix CI and pre-commit
1 parent 8943e0a commit f330473

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

pandas/core/util/numba_.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ def prepare_function_arguments(
150150
return args, kwargs
151151

152152

153-
def extract_numba_options(decorator):
153+
def extract_numba_options(decorator: Callable):
154154
"""
155155
Extract targetoptions from a numba.jit decorator
156156
"""

pandas/tests/apply/test_numba.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import pytest
33

44
from pandas.compat import is_platform_arm
5-
from pandas.core.util.numba_ import extract_numba_options
65
import pandas.util._test_decorators as td
76

87
import pandas as pd
@@ -11,6 +10,7 @@
1110
Index,
1211
)
1312
import pandas._testing as tm
13+
from pandas.core.util.numba_ import extract_numba_options
1414
from pandas.util.version import Version
1515

1616
pytestmark = [td.skip_if_no("numba"), pytest.mark.single_cpu, pytest.mark.skipif()]
@@ -130,10 +130,13 @@ def test_numba_unsupported_dtypes(apply_axis):
130130
df["c"].to_frame().apply(f, engine="numba", axis=apply_axis)
131131

132132

133-
@pytest.mark.parametrize("jit_args", [
134-
{"parallel": True, "nogil": True},
135-
{"parallel": False, "nogil": False},
136-
])
133+
@pytest.mark.parametrize(
134+
"jit_args",
135+
[
136+
{"parallel": True, "nogil": True},
137+
{"parallel": False, "nogil": False},
138+
],
139+
)
137140
def test_extract_numba_options_from_user_decorated_function(jit_args):
138141
extracted = extract_numba_options(numba.jit(**jit_args))
139142
for k, v in jit_args.items():

0 commit comments

Comments
 (0)