Commit f979eb7d authored by Olegs NIKISINS's avatar Olegs NIKISINS
Browse files

Fixed the map_labels in the utils, fixing the unit tests

parent 720dd3ce
Pipeline #26193 passed with stage
in 7 minutes and 59 seconds
......@@ -77,7 +77,7 @@ def map_labels(raw_labels, start_index=0):
for i in range(len(possible_labels)):
l = possible_labels[i]
labels[numpy.where(labels==l)[0]] = i + start_index
labels[numpy.where(labels==l)[0][0]] = i + start_index
# -----
# map back to native int, resolve the problem with dataset concatenation
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment