[Minor] Fix intervention return objects

stanfordnlp · Sep 6, 2024 · 95b74ee · 95b74ee
1 parent d4ca094
commit 95b74ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyvene/models/interventions.py b/pyvene/models/interventions.py
@@ -18,7 +18,7 @@ class InterventionOutput(ModelOutput):
     Output of the IntervenableModel, including original outputs, intervened outputs, and collected activations.
     """
     output: Optional[Any] = None
-    loss: Optional[Any] = None
+    latent: Optional[Any] = None
 
 
 class Intervention(torch.nn.Module):