Update dependency management in pyproject.toml

Removed 'pydantic' and 'lightning' from the main dependencies and added them to the 'dev' and 'train' sections respectively for better organization of project dependencies.

Author: Nathaniel Chen

pyproject.toml

@@ -20,8 +20,6 @@ dependencies = [
     "torchvision",
     "torchinfo",
     "scikit-image",
-    "pydantic",
-    "lightning",
     "omegaconf",
     "scikit-learn",
     "tables",
@@ -38,6 +36,11 @@ dev = [
 
     "ipykernel",
     "ipywidgets",
+
+    "pydantic",
+]
+train = [
+    "lightning",
 ]
 eval = [
     "wavio",