From 7965c54cd46c74bea4786f47ba1ef837552c1f14 Mon Sep 17 00:00:00 2001
From: Tom Tseng <tom.hm.tseng@gmail.com>
Date: Thu, 8 Jan 2026 15:56:12 -0800
Subject: [PATCH] tests: Templatize the import

---
 {{cookiecutter.project_slug}}/tests/test_training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/{{cookiecutter.project_slug}}/tests/test_training.py b/{{cookiecutter.project_slug}}/tests/test_training.py
index cd9d1fb..22fa0e4 100644
--- a/{{cookiecutter.project_slug}}/tests/test_training.py
+++ b/{{cookiecutter.project_slug}}/tests/test_training.py
@@ -1,6 +1,6 @@
 import torch
 
-from project_template.training import NeuralNetwork
+from {{ cookiecutter.project_name }}.training import NeuralNetwork
 
 
 def test_neural_network():