ApolloResearch · danbraunai · May 1, 2025 · May 1, 2025 · May 2, 2025 · May 2, 2025
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -38,11 +38,23 @@
             }
         },
         {
-            "name": "lm",
+            "name": "ss",
             "type": "debugpy",
             "request": "launch",
             "program": "${workspaceFolder}/spd/experiments/lm/lm_decomposition.py",
-            "args": "${workspaceFolder}/spd/experiments/lm/lm_config.yaml",
+            "args": "${workspaceFolder}/spd/experiments/lm/ss_config.yaml",
+            "console": "integratedTerminal",
+            "justMyCode": true,
+            "env": {
+                "PYDEVD_DISABLE_FILE_VALIDATION": "1"
+            }
+        },
+        {
+            "name": "ts",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/spd/experiments/lm/lm_decomposition.py",
+            "args": "${workspaceFolder}/spd/experiments/lm/ts_config.yaml",
             "console": "integratedTerminal",
             "justMyCode": true,
             "env": {

diff --git a/pyproject.toml b/pyproject.toml
@@ -24,7 +24,6 @@ dependencies = [
     "streamlit",
     "streamlit-antd-components",
     "datasets",
-    "simple-stories-train"
 ]
 
 [project.optional-dependencies]

diff --git a/spd/attributions.py b/spd/attributions.py