From d498ab79be076e9d7438b55c65bc5695630a6b24 Mon Sep 17 00:00:00 2001
From: Dilyara Bareeva <dilyara.bareeva@hhi.fraunhofer.de>
Date: Mon, 3 Jun 2024 17:06:54 +0200
Subject: [PATCH] add lightning to deps

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index f616f400..c23bd578 100644
--- 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"]