From 37a934bdb351ebe9231fe7bc2f95c16d129bf845 Mon Sep 17 00:00:00 2001
From: Johan Hansson <39947546+JohanWork@users.noreply.github.com>
Date: Wed, 10 Jan 2024 22:50:05 +0100
Subject: [PATCH] clean up

---
 tests/prompt_strategies/test_alpacha.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tests/prompt_strategies/test_alpacha.py b/tests/prompt_strategies/test_alpacha.py
index 64027740a..cd02062f9 100644
--- a/tests/prompt_strategies/test_alpacha.py
+++ b/tests/prompt_strategies/test_alpacha.py
@@ -1,5 +1,5 @@
 """
-Test module for sharegpt integration w chatml
+Test module for alpacha integration w chatml
 """
 import pytest
 from datasets import Dataset
@@ -7,13 +7,12 @@ from tokenizers import AddedToken
 from transformers import AutoTokenizer
 
 from axolotl.datasets import TokenizedPromptDataset
-from axolotl.prompt_strategies.sharegpt import SimpleShareGPTPromptTokenizingStrategy
 from axolotl.prompt_tokenizers import AlpacaPromptTokenizingStrategy
 from axolotl.prompters import AlpacaPrompter
 
 
 @pytest.fixture(name="alpacha_dataset")
-def fixture_sharegpt_dataset():
+def fixture_alpacha_dataset():
     return Dataset.from_list(
         [
             {"instruction": "Evaluate this sentence for spelling and grammar mistakes", "input": "He finnished his meal and left the resturant", "output": "He finished his meal and left the restaurant."}