fix datasets
1 changed file+4−4
benchcompress/src/benchcompress/datasets/seismic/__init__.pymodified+4−4View file
@@ -60,7 +60,7 @@ def _load_04A_04B_seismic_data() -> np.ndarray:
6060
6161 def _load_quantized_04A_04B_seismic_data():
6262 X = _load_04A_04B_seismic_data()
63- step = 10
63+ step = 10000
6464 X = np.round(X / step).astype(np.int32)
6565 return X
6666
@@ -89,7 +89,7 @@ def _load_lim_2024_seismic_data() -> np.ndarray:
8989
9090 def _load_quantized_lim_2024_seismic_data():
9191 X = _load_lim_2024_seismic_data()
92- step = 10000
92+ step = 10
9393 X = np.round(X / step).astype(np.int32)
9494 return X
9595
@@ -106,7 +106,7 @@ datasets = [
106106 },
107107 {
108108 "name": "seismic-04A-04B-quantized",
109- "version": "2",
109+ "version": "3",
110110 "description": "Seismic data from Roger Revelle voyage RR1508, quantized.",
111111 "create": lambda: _load_quantized_04A_04B_seismic_data(),
112112 "tags": tags_integer,
@@ -124,7 +124,7 @@ datasets = [
124124 },
125125 {
126126 "name": "seismic-lim-2024-01-quantized",
127- "version": "2",
127+ "version": "3",
128128 "description": "Seismic data from the 2022 Mw 3.8 Goesan earthquake in South Korea, quantized.",
129129 "create": lambda: _load_quantized_lim_2024_seismic_data(),
130130 "tags": tags_integer,