From 2194cc7b4fb70b3397d220e09b01f53822003fd6 Mon Sep 17 00:00:00 2001
From: Christian Risi <75698846+CnF-Gris@users.noreply.github.com>
Date: Thu, 2 Oct 2025 09:56:05 +0200
Subject: [PATCH] Changed test to use pool trainer

---
 Project_Model/Tests/bpe_trainer_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Project_Model/Tests/bpe_trainer_test.py b/Project_Model/Tests/bpe_trainer_test.py
index 2e1fa08..69ac4bb 100644
--- a/Project_Model/Tests/bpe_trainer_test.py
+++ b/Project_Model/Tests/bpe_trainer_test.py
@@ -4,13 +4,13 @@ import Project_Model.Libs.BPE as BPE
 
 import re
 
-CACHE_DIR_PATH = Path("Project_Model/Tests/trainer_files/cache")
+CACHE_DIR_PATH = Path("Project_Model/Tests/trainer_files/cache/pool-cache.json")
 
 class TestTrainBPE:
 
     def test_bpe_train_encoding_simple(self):
 
-        TRAINER = BPE.NanoSocraTrainer(
+        TRAINER = BPE.NanoSocraTrainerPool(
             int(32E3),
             ["<SOT>", "<EOT>"],
             40