@@ -114,13 +114,13 @@ def create_cats(self, size, cats_rep, entries=False):
114114 if col .multi_min and col .multi_max :
115115 if HAS_GPU :
116116 ser = dist .create_col (
117- col_size + 1 , dtype = np . long , min_val = col .multi_min , max_val = col .multi_max
117+ col_size + 1 , dtype = int , min_val = col .multi_min , max_val = col .multi_max
118118 )
119119 ser = make_series (np .ceil (ser )).astype (ser .dtype )
120120 _cumsum = xp .cumsum
121121 else :
122122 ser = dist .create_col (
123- col_size + 1 , dtype = np . long , min_val = col .multi_min , max_val = col .multi_max
123+ col_size + 1 , dtype = int , min_val = col .multi_min , max_val = col .multi_max
124124 )
125125 ser = make_df (np .ceil (ser ))[0 ]
126126 _cumsum = np .cumsum
@@ -130,12 +130,12 @@ def create_cats(self, size, cats_rep, entries=False):
130130 offs = offs .astype ("int32" )
131131 if HAS_GPU :
132132 ser = dist .create_col (
133- col_size , dtype = np . long , min_val = col .min_val , max_val = col .cardinality
133+ col_size , dtype = int , min_val = col .min_val , max_val = col .cardinality
134134 )
135135 ser = make_series (np .ceil (ser )).astype (ser .dtype )
136136 else :
137137 ser = dist .create_col (
138- col_size , dtype = np . long , min_val = col .min_val , max_val = col .cardinality
138+ col_size , dtype = int , min_val = col .min_val , max_val = col .cardinality
139139 )
140140 ser = make_df (np .ceil (ser ))[0 ]
141141 ser = ser .astype ("int32" )
0 commit comments