diff --git a/src/data_loading.py b/src/data_loading.py index 9642c7a..2b62f20 100644 --- a/src/data_loading.py +++ b/src/data_loading.py @@ -25,9 +25,9 @@ def __detect_tristan_data_version(file: h5py.File) -> int: If the version of Tristan cannot be identified. """ # The fields to query were chosen only because they don't exist in the other version of Tristan - # Files: particles fields spectra paramater + # Files: particles fields spectra paramater: extra for improved matching on some dataset v1_keys = ('che', 'densi', 'gamma', 'acool') - v2_keys = ('ind_1', 'dens1', 'ebins', 'algorithm:c') + v2_keys = ('ind_1', 'dens1', 'ebins', 'algorithm:c', 'ind_5') if any(key in file for key in v1_keys): return 1