From 27fd56224976852f21c164b500ec9d4de6024508 Mon Sep 17 00:00:00 2001
From: Alex Morehead <alex.morehead@gmail.com>
Date: Fri, 1 Dec 2023 14:49:36 -0600
Subject: [PATCH] Update docstring

---
 src/models/mnist_module.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/models/mnist_module.py b/src/models/mnist_module.py
index 6f613865c..294d5341c 100644
--- a/src/models/mnist_module.py
+++ b/src/models/mnist_module.py
@@ -51,6 +51,7 @@ def __init__(
         :param net: The model to train.
         :param optimizer: The optimizer to use for training.
         :param scheduler: The learning rate scheduler to use for training.
+        :param compile: Whether to compile the model before training.
         """
         super().__init__()