@@ -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+
17141719cdef 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