Fix numpy 1.24 support

This commit is contained in:
allegroai 2022-12-29 13:19:17 +02:00
parent c75c83c21d
commit 0029f131c9

View File

@ -96,7 +96,7 @@ class WeightsGradientHistHelper(object):
cur_idx_above = cur_idx_above[:int(_histogram_granularity * ratio / (1 + ratio))] cur_idx_above = cur_idx_above[:int(_histogram_granularity * ratio / (1 + ratio))]
else: else:
cur_idx_above = np.array([]) cur_idx_above = np.array([])
_cur_idx = np.unique(np.sort(np.concatenate((cur_idx_below, cur_idx_above)).astype(np.int))) _cur_idx = np.unique(np.sort(np.concatenate((cur_idx_below, cur_idx_above)).astype(np.int64)))
return _cur_idx return _cur_idx
def add_histograms(self, histograms): def add_histograms(self, histograms):