brain-score.org submission (user:121) | (public:False)

brainscore_language/models/test_embedding_4/__init__.py

@@ -0,0 +1,17 @@
+"""
+Test model for workflow testing.
+
+This is a minimal, fast-running model designed for testing the plugin submission workflow.
+It uses distilgpt2 from HuggingFace, similar to the gpt models.
+"""
+from brainscore_language import model_registry
+from brainscore_language import ArtificialSubject
+from brainscore_language.model_helpers.huggingface import HuggingfaceSubject
+
+# Register the test model using distilgpt2 (smallest/fastest GPT model)
+model_registry['test_embedding_4'] = lambda: HuggingfaceSubject(
+    model_id='distilgpt2',
+    region_layer_mapping={
+        ArtificialSubject.RecordingTarget.language_system: 'transformer.h.5'
+    }
+)

brainscore_language/models/test_embedding_4/test.py

@@ -0,0 +1,17 @@
+"""
+Tests for the test-embedding model.
+"""
+import numpy as np
+import pytest
+from brainscore_language import load_model, ArtificialSubject, score
+
+
+def test_neural():
+    """Test that the model can produce neural recordings."""
+    assert "Mike" == "Mike"
+
+
+def test_score():
+    """Test that the model can be scored on a benchmark."""
+    # Use a small, fast benchmark for testing
+    assert True