diff --git a/requirements.txt b/requirements.txt
index 4b4eeb928ada46997ff633014b4c8f41f4853665..ce3c43f8365968edeee1ba51151bf10c8532b4f2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,5 +14,4 @@ plotly~=5.8.0
 beautifulsoup4~=4.11.1
 lxml~=4.9.1
 torch
-tensorboardX
 tqdm
\ No newline at end of file
diff --git a/setup.py b/setup.py
index f2d6499cf3ee022155f915ab2c4ece7a9a16fb48..a827ce14a8c425ea2095437f67ef0339dd7d9f3b 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,9 @@ setup(
         "igraph~=0.9.10",
         "plotly~=5.8.0",
         "beautifulsoup4~=4.11.1",
-        "lxml~=4.9.1"
+        "lxml~=4.9.1",
+        "torch",
+        "tqdm"
     ],
     classifiers=[
         'Development Status :: 2 - Pre-Alpha',