add lightning to deps

dilyabareeva · Jun 3, 2024 · d498ab7 · d498ab7
1 parent 069458d
commit d498ab7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,6 +10,7 @@ dependencies = [
     "numpy>=1.19.5",
     "torch>=1.13.1",
     "captum>=0.6.0",
+    "lighning>=1.4.0",
 ]
 dynamic = ["version"]