update setup file

Author: BradleyEdelman

README.md

@@ -38,7 +38,7 @@ This version introduces a **refined adaptive training strategy with a constant p
 You can install the latest version of EdgeTrain via pip:
 
 ```bash
-pip install https://github.com/BradleyEdelman/EdgeTrain/releases/download/v0.2.0-alpha/edgetrain-0.2.0.tar.gz
+pip install https://github.com/BradleyEdelman/EdgeTrain/releases/download/v0.2.0/edgetrain-0.2.0.tar.gz
 ```
 
 Alternatively, clone the repository and install manually:
@@ -103,7 +103,7 @@ EdgeTrain/
 │
 ├── .github/workflows/
 │   ├── ci.yml
-│   └──lint.yml
+│   └── lint.yml
 │
 ├── .flake8
 ├── .gitignore

setup.py

@@ -2,21 +2,29 @@
 
 setup(
     name="edgetrain",
-    version="0.1.1-alpha",
+    version="0.2.0",
     packages=find_packages(),
     install_requires=[
-        "tensorflow>=2.0.0",
         "psutil>=5.0.0",
         "GPUtil>=1.4.0",
         "matplotlib>=3.7.0",
         "pandas>=1.5.0",
-        "numpy>=1.24.0",
         "pynvml>=8.0.0",
-        "torch>=2.5.1",
+        "tensorflow==2.12.0",
+        "keras==2.12.0",
+        "tensorflow-model-optimization==0.7.3",
+        "jupyter",
     ],
     extras_require={
         "dev": [
-            "pytest",  # for testing
+            "pytest",
+            "pytest-cov",
+            "pytest-mock",
+            "pre-commit",
+            "black",
+            "black[jupyter]",
+            "flake8",
+            "isort",
         ]
     },
     description="A utility for machine learning training with limited resources.",