From d2829e27fa4b81a4eaae0bb98d1ba834aaa10276 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ga=C5=A1per=20Spagnolo?= <gasper.spagnolo@outlook.com>
Date: Sun, 19 Mar 2023 11:23:11 +0100
Subject: [PATCH] Update requirements.txt

---
 README.md             | 6 ++++++
 code/.gitignore       | 1 +
 code/requirements.txt | 3 +--
 3 files changed, 8 insertions(+), 2 deletions(-)
 create mode 100644 code/.gitignore

diff --git a/README.md b/README.md
index 43475e4..90385e9 100644
--- a/README.md
+++ b/README.md
@@ -7,3 +7,9 @@ cd ./code/ && python -m venv .venv
 . ./.venv/bin/activate
 pip install -r requirements.txt
 ```
+
+### Lint
+
+```bash
+black ./code
+```
diff --git a/code/.gitignore b/code/.gitignore
new file mode 100644
index 0000000..691037e
--- /dev/null
+++ b/code/.gitignore
@@ -0,0 +1 @@
+.venv/*
diff --git a/code/requirements.txt b/code/requirements.txt
index 09a1412..d007f77 100644
--- a/code/requirements.txt
+++ b/code/requirements.txt
@@ -1,3 +1,4 @@
+--extra-index-url=https://download.pytorch.org/whl/cu118/
 black==23.1.0
 certifi==2022.12.7
 charset-normalizer==2.1.1
@@ -18,7 +19,6 @@ mpmath==1.2.1
 mypy-extensions==1.0.0
 networkx==3.0
 numpy==1.24.1
-opencv-python==4.7.0.72
 packaging==23.0
 pandas==1.5.3
 pathspec==0.11.1
@@ -29,7 +29,6 @@ pyparsing==3.0.9
 python-dateutil==2.8.2
 pytz==2022.7.1
 requests==2.28.1
-seaborn==0.12.2
 six==1.16.0
 sympy==1.11.1
 tomli==2.0.1