Skip to content

Commit 43ca219

Browse files
committed
Fixed __norm__ error in batch_generator
1 parent dfed17e commit 43ca219

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

scripts/batch_generator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -253,8 +253,9 @@ def get_colidxs_from_colnames(data, columns):
253253
if config.scale_field == '__norm__':
254254
self._normalizer_idx = None
255255
else:
256-
self._normalizer_idx = np_array_index(colnames, config.scale_field)
257-
256+
#self._normalizer_idx = np_array_index(colnames, config.scale_field)
257+
self._normalizer_idx = colnames.index(config.scale_field)
258+
258259
# Set up input-related attributes
259260
self._num_inputs = config.num_inputs = len(self._feature_names)
260261

0 commit comments

Comments
 (0)