fix bug in data_utils.py for Issue #10

xlxwalex · Mar 7, 2023 · 8cd55a2 · 8cd55a2
1 parent 3c6dd37
commit 8cd55a2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/model/STG-correction/utils/data_utils.py b/model/STG-correction/utils/data_utils.py
@@ -425,13 +425,15 @@ def convert_spmap_sw(sp_map : list, sw_labels : np.array) -> list:
 def convert_spmap_tg(sp_map : list, tg_mapper : list) -> list:
  new_spmap = []
  for idx in range(len(sp_map)):
+ if idx == 163:
+ print()
  spmapper = sp_map[idx]
  tgmapper = tg_mapper[idx]
  mapper = {}
  if not spmapper: new_spmap.append(None)
  else:
  for sp in spmapper:
- if sp in tgmapper and tgmapper[sp] > 0: mapper[tgmapper[sp]] = spmapper[sp]
+ if sp in tgmapper and tgmapper[sp] >= 0: mapper[tgmapper[sp]] = spmapper[sp]
  new_spmap.append(mapper)
  return new_spmap