update names
6 changed files+20−18
python/ephys_compression_tests/datasets/__init__.pymodified+4−4View file
@@ -1,10 +1,10 @@
1-from .vyom import datasets as vyom_datasets
2-from .aind import datasets as aind_datasets
1+from .retina512 import datasets as retina512_datasets
2+from .aind_compression import datasets as aind_compression_datasets
33 from ..types import Dataset
44
55 datasets_list = [
6- vyom_datasets,
7- aind_datasets,
6+ retina512_datasets,
7+ aind_compression_datasets,
88 ]
99
1010 datasets: list[Dataset] = []
python/ephys_compression_tests/datasets/aind/aind.mddeleted+0−1View file
@@ -1 +0,0 @@
1-Examples from AIND
python/ephys_compression_tests/datasets/aind/__init__.py →python/ephys_compression_tests/datasets/aind_compression/__init__.pyrenamed+4−4View file
@@ -7,12 +7,12 @@ from ...types import Dataset
77 from ..._filters import bandpass_filter
88
99
10-SOURCE_FILE = "aind/__init__.py"
10+SOURCE_FILE = "aind-compression/__init__.py"
1111
1212
1313 def _load_long_description():
1414 current_dir = os.path.dirname(os.path.abspath(__file__))
15- md_path = os.path.join(current_dir, "aind.md")
15+ md_path = os.path.join(current_dir, "aind-compression.md")
1616 with open(md_path, "r", encoding="utf-8") as f:
1717 return f.read()
1818
@@ -39,7 +39,7 @@ def load_aind_ch101() -> np.ndarray:
3939
4040 dataset_dicts_base = [
4141 {
42- "name": "aind-ch101",
42+ "name": "aind-compression-np2-ProbeB-ch101",
4343 "version": "1",
4444 "description": "AIND CH101 dataset",
4545 "create": load_aind_ch101,
@@ -63,7 +63,7 @@ for d in dataset_dicts_base:
6363
6464 dataset_dicts.append(
6565 {
66- "name": f'{d["name"]}-bandpass',
66+ "name": f'{d["name"]}-filtered',
6767 "version": "1",
6868 "description": f'{d["description"]} (bandpass filtered 300-4000 Hz)',
6969 "create": create0,
python/ephys_compression_tests/datasets/aind_compression/aind-compression.mdadded+3−0View file
@@ -0,0 +1,3 @@
1+Examples from AIND Compression Benchmark
2+
3+See: https://allenneuraldynamics.github.io/data.html#aind-benchmark-dataephys-compression
python/ephys_compression_tests/datasets/vyom/__init__.py →python/ephys_compression_tests/datasets/retina512/__init__.pyrenamed+9−9View file
@@ -7,12 +7,12 @@ from ...types import Dataset
77 from ..._filters import bandpass_filter
88
99
10-SOURCE_FILE = "vyom/__init__.py"
10+SOURCE_FILE = "retina512/__init__.py"
1111
1212
1313 def _load_long_description():
1414 current_dir = os.path.dirname(os.path.abspath(__file__))
15- md_path = os.path.join(current_dir, "vyom.md")
15+ md_path = os.path.join(current_dir, "retina512.md")
1616 with open(md_path, "r", encoding="utf-8") as f:
1717 return f.read()
1818
@@ -22,14 +22,14 @@ LONG_DESCRIPTION = _load_long_description()
2222 tags = ["real", "ecephys", "timeseries", "1d", "integer", "correlated"]
2323
2424
25-def load_vyom_example_ch0_seg2_6() -> np.ndarray:
26- """Load Vyom example dataset from external URL.
25+def load_retina512_example_ch0_seg2_6() -> np.ndarray:
26+ """Load Retina512 example dataset from external URL.
2727
2828 Returns:
2929 Array containing the loaded data
3030 """
3131 url = "https://tempory.net/ephys-compression-tests/vyom_example_ch0_seg2-6.npy"
32- print(f'Loading Vyom example dataset from {url}...')
32+ print(f'Loading Retina512 example dataset from {url}...')
3333 response = requests.get(url)
3434 response.raise_for_status()
3535 data = np.load(io.BytesIO(response.content))
@@ -39,10 +39,10 @@ def load_vyom_example_ch0_seg2_6() -> np.ndarray:
3939
4040 dataset_dicts_base = [
4141 {
42- "name": "vyom-example-ch0-seg2-6",
42+ "name": "retina512-ch0-seg2-6",
4343 "version": "1",
44- "description": "Vyom example dataset",
45- "create": load_vyom_example_ch0_seg2_6,
44+ "description": "Retina512 example dataset",
45+ "create": load_retina512_example_ch0_seg2_6,
4646 "tags": tags,
4747 "source_file": SOURCE_FILE,
4848 "long_description": LONG_DESCRIPTION,
@@ -63,7 +63,7 @@ for d in dataset_dicts_base:
6363
6464 dataset_dicts.append(
6565 {
66- "name": f'{d["name"]}-bandpass',
66+ "name": f'{d["name"]}-filtered',
6767 "version": "1",
6868 "description": f'{d["description"]} (bandpass filtered 300-4000 Hz)',
6969 "create": create0,
python/ephys_compression_tests/datasets/vyom/vyom.md →python/ephys_compression_tests/datasets/retina512/retina512.mdrenamed+0−0View file
No changes to the file's content.