From 046e36678e937aaa270da7b14510657515ae7cda Mon Sep 17 00:00:00 2001 From: Craig Date: Sat, 12 Apr 2025 12:01:27 +0100 Subject: [PATCH] Pausing for now, train loop should now work and adding some tests --- tests/test_visualization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_visualization.py b/tests/test_visualization.py index 4eb51ee..e66f27b 100644 --- a/tests/test_visualization.py +++ b/tests/test_visualization.py @@ -6,7 +6,7 @@ import torch # Import visualization functions sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) -from scripts.visualize_predictions import visualize_prediction +from scripts.visualize_predictions import visualize_prediction # noqa: E402 def test_visualize_prediction():