craig/torchvision-vibecoding-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Pausing for now, train loop should now work and adding some tests

Browse Source
This commit is contained in:
Craig
2025-04-12 12:01:27 +01:00
parent be70c4e160
commit 046e36678e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_visualization.py
View File

@@ -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():