Added dependencies to pyproject

Author: soupault

metko/_mutual_info.py

@@ -1,7 +1,7 @@
-from PIL import Image
+# from PIL import Image
 import torch
 from torch import nn
-from torchvision import transforms
+# from torchvision import transforms
 
 
 # Modified from https://github.com/connorlee77/pytorch-mutual-information

pyproject.toml

@@ -16,14 +16,15 @@ classifiers = [
     "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
 ]
-#dependencies = [
-#    "numpy",
-#    "scipy",
-#    "scikit-learn",
-#    "matplotlib",
-#    "numba",
-#    "pytorch",
-#]
+dependencies = [
+    "numpy",
+    "scipy",
+    "matplotlib",
+    "seaborn",
+    "numba",
+    "torch",
+    "torchvision",
+]
 
 [project.urls]
 "Homepage" = "https://github.com/soupault/metko"