Commit 8374931b authored by Martino Bertoni's avatar Martino Bertoni 🌋
Browse files

bugfix: neig.get_vector better return string inchikeys rather than bytes

parent d6d779c0
Pipeline #2595 passed with stages
in 18 minutes and 38 seconds
......@@ -277,10 +277,10 @@ class neig(BaseSignature, DataSignature):
inks, signs = list(), list()
with h5py.File(self.data_path, 'r') as hf:
dset = hf[dataset_name]
col_keys = hf['col_keys'][:]
col_keys = hf['col_keys'][:].astype(str)
dset_shape = dset.shape
for idx in sorted(idxs.flatten()):
inks.append(self.row_keys[idx])
inks.append(self.row_keys[idx].astype(str))
if dataset_name == 'indices':
signs.append(col_keys[dset[idx]])
else:
......
Markdown is supported
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