add np1 dataset
1 changed file+21−4
python/ephys_compression_tests/datasets/aind_compression/__init__.pymodified+21−4View file
@@ -22,7 +22,7 @@ LONG_DESCRIPTION = _load_long_description()
2222 tags = ["real", "ecephys", "timeseries", "integer", "correlated"]
2323
2424
25-def load_aind_ch101() -> np.ndarray:
25+def load_aind_np2_probeB_ch101() -> np.ndarray:
2626 """Load AIND CH101 dataset from external URL.
2727
2828 Returns:
@@ -35,7 +35,7 @@ def load_aind_ch101() -> np.ndarray:
3535 data = np.load(io.BytesIO(response.content)).flatten()
3636 return data
3737
38-def load_aind_ch101_110() -> np.ndarray:
38+def load_aind_np2_probeB_ch101_110() -> np.ndarray:
3939 url = "https://tempory.net/ephys-compression-tests/aind/aind_compression_np2_probeB_ch101-110.raw.npy"
4040 print(f'Loading AIND dataset from {url}...')
4141 response = requests.get(url)
@@ -43,12 +43,20 @@ def load_aind_ch101_110() -> np.ndarray:
4343 data = np.load(io.BytesIO(response.content))
4444 return data
4545
46+def load_aind_np1_probeA_101_110() -> np.ndarray:
47+ url = "https://tempory.net/ephys-compression-tests/aind/aind-np1-probeA-ch101-110.raw.npy"
48+ print(f'Loading AIND dataset from {url}...')
49+ response = requests.get(url)
50+ response.raise_for_status()
51+ data = np.load(io.BytesIO(response.content))
52+ return data
53+
4654 dataset_dicts_base = [
4755 {
4856 "name": "aind-compression-np2-ProbeB-ch101",
4957 "version": "1",
5058 "description": "AIND CH101 dataset",
51- "create": load_aind_ch101,
59+ "create": load_aind_np2_probeB_ch101,
5260 "tags": tags + ["single-channel"],
5361 "source_file": SOURCE_FILE,
5462 "long_description": LONG_DESCRIPTION,
@@ -57,7 +65,16 @@ dataset_dicts_base = [
5765 "name": "aind-compression-np2-ProbeB-ch101-110",
5866 "version": "1",
5967 "description": "AIND CH101-110 dataset",
60- "create": load_aind_ch101_110,
68+ "create": load_aind_np2_probeB_ch101_110,
69+ "tags": tags + ["multi-channel"],
70+ "source_file": SOURCE_FILE,
71+ "long_description": LONG_DESCRIPTION,
72+ },
73+ {
74+ "name": "aind-compression-np1-ProbeA-ch101-110",
75+ "version": "1",
76+ "description": "AIND NP1 ProbeA CH101-110 dataset",
77+ "create": load_aind_np1_probeA_101_110,
6178 "tags": tags + ["multi-channel"],
6279 "source_file": SOURCE_FILE,
6380 "long_description": LONG_DESCRIPTION,