diff --git a/brainscore_vision/data/DataGenerationTrial_emogan/__init__.py b/brainscore_vision/data/DataGenerationTrial_emogan/__init__.py new file mode 100644 index 000000000..43ae9d6aa --- /dev/null +++ b/brainscore_vision/data/DataGenerationTrial_emogan/__init__.py @@ -0,0 +1,25 @@ +from brainio.assemblies import NeuronRecordingAssembly +from brainscore_vision import load_stimulus_set +from brainscore_vision import stimulus_set_registry, data_registry +from brainscore_vision.data_helpers.s3 import load_assembly_from_s3, load_stimulus_set_from_s3 + +stimulus_set_registry["DataGenerationTrial_emogan"] = lambda: load_stimulus_set_from_s3( + identifier="DataGenerationTrial_emogan", + bucket="brainio-brainscore", + csv_version_id="uaMuF5sFgQzj2WTDHiCDuX5Zzb13M3as", + csv_sha1="a26c9089672837dd7f5186c7c42c84d9805a3965", + zip_version_id="flAOeJG4aLNcdEErZ5XhI1uW3ucwqVcS", + zip_sha1="dd39ddb1fd053151793e3f9f6e85b664cc4636bc", + filename_prefix='stimulus_', +) + +data_registry["DataGenerationTrial_emogan"] = lambda: load_assembly_from_s3( + identifier="DataGenerationTrial_emogan", + version_id="TJ4EIBPvnpxITkljkkjNNsAxdRDvRPqr", + sha1="14a331c49da73ee1838e9b5e85c76073288f601e", + bucket="brainio-brainscore", + cls=NeuronRecordingAssembly, + stimulus_set_loader=lambda: load_stimulus_set('DataGenerationTrial_emogan'), +) + + \ No newline at end of file diff --git a/brainscore_vision/data/DataGenerationTrial_emogan/data_packaging.py b/brainscore_vision/data/DataGenerationTrial_emogan/data_packaging.py new file mode 100644 index 000000000..74b42cfe2 --- /dev/null +++ b/brainscore_vision/data/DataGenerationTrial_emogan/data_packaging.py @@ -0,0 +1,19 @@ +from brainio.packaging import package_stimulus_set, package_data_assembly +from brainscore_vision import load_dataset, load_stimulus_set + +def upload_stimulus_set_to_s3(stimuli): + return package_stimulus_set(catalog_name=None, proto_stimulus_set=stimuli, + stimulus_set_identifier=stimuli.name, bucket_name="brainio-brainscore") + +def upload_assembly_to_s3(assembly): + return package_data_assembly(None, assembly, assembly_identifier=assembly.name, + stimulus_set_identifier=assembly.name, + assembly_class_name="NeuronRecordingAssembly", + bucket_name="brainio-brainscore") + + +if __name__ == '__main__': + ss = load_stimulus_set('DataGenerationTrial_emogan') + assembly = load_dataset('DataGenerationTrial_emogan') + + \ No newline at end of file diff --git a/brainscore_vision/data/DataGenerationTrial_emogan/test.py b/brainscore_vision/data/DataGenerationTrial_emogan/test.py new file mode 100644 index 000000000..bb4de5626 --- /dev/null +++ b/brainscore_vision/data/DataGenerationTrial_emogan/test.py @@ -0,0 +1,23 @@ +import pytest + +from brainscore_vision import load_dataset, load_stimulus_set +from brainscore_vision.benchmark_helpers import check_standard_format + + +@pytest.mark.memory_intense +@pytest.mark.private_access +def test_assembly(): + assembly = load_dataset('DataGenerationTrial_emogan') + check_standard_format(assembly, nans_expected=True) + assert assembly.attrs['stimulus_set_identifier'] == 'DataGenerationTrial_emogan' + assert set(assembly['region'].values) == {'IT'} + assert len(assembly['presentation']) == 8500 + assert len(assembly['neuroid']) == 18 + + +@pytest.mark.private_access +def test_stimulus_set(): + stimulus_set = load_stimulus_set('DataGenerationTrial_emogan') + assert len(stimulus_set) == 170 + + \ No newline at end of file