Update requirements.txt

README.md

@@ -7,3 +7,9 @@ cd ./code/ && python -m venv .venv
 . ./.venv/bin/activate
 pip install -r requirements.txt
 ```
+
+### Lint
+
+```bash
+black ./code
+```

code/.gitignore

@@ -0,0 +1 @@
+.venv/*

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