diff --git a/Project_Model/Libs/BPE/Classes/__init__.py b/Project_Model/Libs/BPE/Classes/__init__.py index 32e958a..d3b93b6 100644 --- a/Project_Model/Libs/BPE/Classes/__init__.py +++ b/Project_Model/Libs/BPE/Classes/__init__.py @@ -3,11 +3,13 @@ from .NanoSocratesSplitter import NanoSocratesSplitter from .NanoSocratesBPE import NanoSocratesBPE, NanoSocratesBatchMemoryBPE from .NanoSocraTrainer import NanoSocraTrainer from .NanoSocraTraineRam import NanoSocraTraineRam +from .NanoSocraTrainerPool import NanoSocraTrainerPool __all__ = [ "NanoSocratesChunker", "NanoSocratesSplitter", "NanoSocratesBPE", "NanoSocraTrainer", - "NanoSocraTraineRam" + "NanoSocraTraineRam", + "NanoSocraTrainerPool" ] \ No newline at end of file