From d60abb07a6fe8c380391214b913b875c8aec5747 Mon Sep 17 00:00:00 2001
From: Vedant Puri <vedantpuri@gmail.com>
Date: Sun, 10 Sep 2023 16:16:36 -0400
Subject: [PATCH] comment

---
 src/GeometryLearning.jl | 1 -
 src/train.jl            | 3 ++-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/GeometryLearning.jl b/src/GeometryLearning.jl
index 95112568..1c77764c 100644
--- a/src/GeometryLearning.jl
+++ b/src/GeometryLearning.jl
@@ -4,7 +4,6 @@ using DocStringExtensions
 
 # PDE stack
 using FourierSpaces
-using FourierSpaces: linspace
 
 # ML Stack
 using Lux
diff --git a/src/train.jl b/src/train.jl
index 1465565d..6b04c814 100644
--- a/src/train.jl
+++ b/src/train.jl
@@ -121,7 +121,8 @@ function train_model(
         cb_stats(p, st)
     end
 
-    # TODO: output a train.log file with timings
+    # TODO - output a train.log file with timings
+    # add ProgressMeters.jl
 
     # save statistics
     statsfile = open(joinpath(dir, "statistics.txt"), "w")