From 797cfb20de82b1c0183c9ffd08f16f95d4c3f4e8 Mon Sep 17 00:00:00 2001
From: rasbt <mail@sebastianraschka.com>
Date: Fri, 29 Mar 2024 09:03:36 -0500
Subject: [PATCH] fix test

---
 ch05/01_main-chapter-code/tests.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ch05/01_main-chapter-code/tests.py b/ch05/01_main-chapter-code/tests.py
index dea2300..e410169 100644
--- a/ch05/01_main-chapter-code/tests.py
+++ b/ch05/01_main-chapter-code/tests.py
@@ -6,7 +6,7 @@
 # File for internal use (unit tests)
 
 import pytest
-from train import main
+from gpt_train import main
 
 
 @pytest.fixture