cleanup and support files

.gitignore

@@ -4,3 +4,11 @@
 
 # Environment
 .venv
+
+# Packaging
+.eggs
+*.egg
+*.egg-info
+
+# Python
+*__pycache__

mypy.ini

@@ -0,0 +1,3 @@
+[mypy]
+python_version = 3.7
+show_error_codes = True

pyproject.toml

@@ -0,0 +1,13 @@
+
+[tool.pycln]
+all = true
+
+[tool.mypy]
+ignore_missing_imports = true
+
+[tool.isort]
+multi_line_output = 3
+include_trailing_comma = true
+force_grid_wrap = 0
+use_parentheses = true
+ensure_newline_before_comments = true

requirements-train.txt

@@ -0,0 +1,5 @@
+torch
+wandb
+addict
+scipy
+pytest

setup.py

@@ -38,8 +38,8 @@ def read_requirements(path):
     long_description_content_type="text/markdown",
     author="Antonio Stanziola, UCL BUG",
     author_email="[email protected]",
-    packages=find_packages(exclude=["tests", ".github", "docs"]),
-    python_requires=">=3.7",
+    packages=find_packages(exclude=["data", ".github", "scripts"]),
+    python_requires=">=3.8",
     install_requires=read_requirements("requirements.txt"),
     license="MIT License"
 )