Skip to content

Commit 322adfc

Browse files
committed
further fixes
1 parent 478a70f commit 322adfc

File tree

1 file changed

+17
-12
lines changed

1 file changed

+17
-12
lines changed

src/rapidfuzz/process_cpp_impl.pyx

Lines changed: 17 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -941,8 +941,8 @@ cdef inline extract_dict_size_t(
941941

942942
cdef vector[DictMatchElem[size_t]] results = extract_dict_impl[size_t](
943943
scorer_kwargs, scorer_flags, scorer, proc_query, proc_choices,
944-
get_score_cutoff_i64(score_cutoff, scorer_flags),
945-
get_score_cutoff_i64(score_hint, scorer_flags)
944+
get_score_cutoff_size_t(score_cutoff, scorer_flags),
945+
get_score_cutoff_size_t(score_hint, scorer_flags)
946946
)
947947

948948
# due to score_cutoff not always completely filled
@@ -1086,8 +1086,8 @@ cdef inline extract_list_size_t(
10861086

10871087
cdef vector[ListMatchElem[size_t]] results = extract_list_impl[size_t](
10881088
scorer_kwargs, scorer_flags, scorer, proc_query, proc_choices,
1089-
get_score_cutoff_i64(score_cutoff, scorer_flags),
1090-
get_score_cutoff_i64(score_hint, scorer_flags)
1089+
get_score_cutoff_size_t(score_cutoff, scorer_flags),
1090+
get_score_cutoff_size_t(score_hint, scorer_flags)
10911091
)
10921092

10931093
# due to score_cutoff not always completely filled
@@ -1711,6 +1711,11 @@ cdef inline MatrixType dtype_to_type_num_i64(dtype) except MatrixType.UNDEFINED:
17111711
return MatrixType.INT32
17121712
return <MatrixType>dtype
17131713

1714+
cdef inline MatrixType dtype_to_type_num_size_t(dtype) except MatrixType.UNDEFINED:
1715+
if dtype is None:
1716+
return MatrixType.UINT32
1717+
return <MatrixType>dtype
1718+
17141719
cdef inline MatrixType dtype_to_type_num_py(dtype, scorer, dict scorer_kwargs) except MatrixType.UNDEFINED:
17151720
import numpy as np
17161721

@@ -1789,12 +1794,12 @@ cdef Matrix cdist_two_lists(
17891794
matrix.matrix = cdist_two_lists_impl[size_t](
17901795
scorer_flags,
17911796
scorer_kwargs, scorer, proc_queries, proc_choices,
1792-
dtype_to_type_num_i64(dtype),
1797+
dtype_to_type_num_size_t(dtype),
17931798
c_workers,
1794-
get_score_cutoff_i64(score_cutoff, scorer_flags),
1795-
get_score_cutoff_i64(score_hint, scorer_flags),
1799+
get_score_cutoff_size_t(score_cutoff, scorer_flags),
1800+
get_score_cutoff_size_t(score_hint, scorer_flags),
17961801
<size_t>score_multiplier,
1797-
scorer_flags.worst_score.i64
1802+
scorer_flags.worst_score.sizet
17981803
)
17991804
elif flags & RF_SCORER_FLAG_RESULT_I64:
18001805
matrix.matrix = cdist_two_lists_impl[int64_t](
@@ -1843,12 +1848,12 @@ cdef Matrix cdist_single_list(
18431848
matrix.matrix = cdist_single_list_impl[size_t](
18441849
scorer_flags,
18451850
scorer_kwargs, scorer, proc_queries,
1846-
dtype_to_type_num_i64(dtype),
1851+
dtype_to_type_num_size_t(dtype),
18471852
c_workers,
1848-
get_score_cutoff_i64(score_cutoff, scorer_flags),
1849-
get_score_cutoff_i64(score_hint, scorer_flags),
1853+
get_score_cutoff_size_t(score_cutoff, scorer_flags),
1854+
get_score_cutoff_size_t(score_hint, scorer_flags),
18501855
<size_t>score_multiplier,
1851-
scorer_flags.worst_score.i64
1856+
scorer_flags.worst_score.sizet
18521857
)
18531858
elif flags & RF_SCORER_FLAG_RESULT_I64:
18541859
matrix.matrix = cdist_single_list_impl[int64_t](

0 commit comments

Comments
 (0)