add wandb

Author: priyakasimbeg

pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
   "clu==0.0.12",
   "matplotlib>=3.9.2",
   "tabulate==0.9.0",
-
+  "wandb==0.21.0"
 ]
 
 [build-system]
@@ -81,8 +81,6 @@ dev = [
   "pre-commit==4.0.1",
 ]
 
-wandb = ["wandb==0.19.6"]
-
 # Workloads
 criteo1tb = ["scikit-learn==1.5.2"]
 fastmri = ["h5py==3.12.0", "scikit-image==0.24.0"]