File tree Expand file tree Collapse file tree 2 files changed +9
-6
lines changed Expand file tree Collapse file tree 2 files changed +9
-6
lines changed Original file line number Diff line number Diff line change @@ -150,7 +150,7 @@ def prepare_function_arguments(
150
150
return args , kwargs
151
151
152
152
153
- def extract_numba_options (decorator ):
153
+ def extract_numba_options (decorator : Callable ):
154
154
"""
155
155
Extract targetoptions from a numba.jit decorator
156
156
"""
Original file line number Diff line number Diff line change 2
2
import pytest
3
3
4
4
from pandas .compat import is_platform_arm
5
- from pandas .core .util .numba_ import extract_numba_options
6
5
import pandas .util ._test_decorators as td
7
6
8
7
import pandas as pd
11
10
Index ,
12
11
)
13
12
import pandas ._testing as tm
13
+ from pandas .core .util .numba_ import extract_numba_options
14
14
from pandas .util .version import Version
15
15
16
16
pytestmark = [td .skip_if_no ("numba" ), pytest .mark .single_cpu , pytest .mark .skipif ()]
@@ -130,10 +130,13 @@ def test_numba_unsupported_dtypes(apply_axis):
130
130
df ["c" ].to_frame ().apply (f , engine = "numba" , axis = apply_axis )
131
131
132
132
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
+ )
137
140
def test_extract_numba_options_from_user_decorated_function (jit_args ):
138
141
extracted = extract_numba_options (numba .jit (** jit_args ))
139
142
for k , v in jit_args .items ():
You can’t perform that action at this time.
0 commit comments