From 1abcd544718e3726cad7b9fe840c98765cba5806 Mon Sep 17 00:00:00 2001
From: "Cyril Zakka, MD" <1841186+cyrilzakka@users.noreply.github.com>
Date: Fri, 16 Feb 2024 17:10:25 -0800
Subject: [PATCH] Fixes imports in train.py

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index 10d4c2cd..5748aed9 100644
--- a/train.py
+++ b/train.py
@@ -4,8 +4,8 @@
 """
 
 # feel free to use either
-from bpe_regex import Tokenizer as RegexTokenizer
-from bpe_basic import Tokenizer as BasicTokenizer
+from bpe_regex import RegexTokenizer
+from bpe_basic import BasicTokenizer
 
 # open some text and train a vocab of 512 tokens
 text = open("taylorswift.txt", "r", encoding="utf-8").read()