format + also bring in the right dependencies for testing

Author: richard-lane

pyproject.toml

@@ -35,6 +35,7 @@ test = [
     "numpy>=2.2.6",
     "pytest>=8.4.1",
     "scikit-image>=0.25.2",
+    "matplotlib>=3.10.3",
 ]
 
 # Formatting, linting, etc.

src/current_denoising/generation/dcgan.py

@@ -154,7 +154,9 @@ def train(
     # We'll be tracking training using the FID score
     fid_model = resnet18(pretrained=True)
     fid_model.to("cuda")
-    fid_metric = FrechetInceptionDistance(feature_dim=1000, model=fid_model, device="cuda")
+    fid_metric = FrechetInceptionDistance(
+        feature_dim=1000, model=fid_model, device="cuda"
+    )
     fid_scores = []
 
     gen_losses = []