diff --git a/sandboxes/columnar_torch.txt b/sandboxes/columnar_torch.txt
index 1fd0573..e883845 100644
--- a/sandboxes/columnar_torch.txt
+++ b/sandboxes/columnar_torch.txt
@@ -3,3 +3,5 @@
 -r ../modules/columnflow/sandboxes/columnar.txt
 
 torch~=2.5.1
+lightning~=2.4.0
+torchmetrics~=1.6.0