@@ -4,6 +4,7 @@ from tests.unit_tests.factories_test import FactoriesTest
 from tests.unit_tests.optimizer_params_override_test import TrainOptimizerParamsOverride
 from tests.unit_tests.resume_training_test import ResumeTrainingTest
 from tests.unit_tests.strictload_enum_test import StrictLoadEnumTest
+from tests.unit_tests.train_after_test_test import CallTrainAfterTestTest
 from tests.unit_tests.zero_weight_decay_on_bias_bn_test import ZeroWdForBnBiasTest
 from tests.unit_tests.save_ckpt_test import SaveCkptListUnitTest
 from tests.unit_tests.all_architectures_test import AllArchitecturesTest
@@ -41,4 +42,5 @@ __all__ = [
     "TrainOptimizerParamsOverride",
     "CallTrainTwiceTest",
     "ResumeTrainingTest",
+    "CallTrainAfterTestTest",
 ]