@@ -22,7 +22,7 @@ from tests.unit_tests.conv_bn_relu_test import TestConvBnRelu
 from tests.unit_tests.initialize_with_dataloaders_test import InitializeWithDataloadersTest
 from tests.unit_tests.training_params_factory_test import TrainingParamsTest
 from tests.unit_tests.config_inspector_test import ConfigInspectTest
-
+from tests.unit_tests.transforms_test import TestTransforms
 
 __all__ = [
     "CrashTipTest",
@@ -48,4 +48,5 @@ __all__ = [
     "ResumeTrainingTest",
     "CallTrainAfterTestTest",
     "ConfigInspectTest",
+    "TestTransforms",
 ]