super-gradients/tests/unit_tests/__init__.py

# PACKAGE IMPORTS FOR EXTERNAL USAGE
from tests.unit_tests.crash_tips_test import CrashTipTest
from tests.unit_tests.double_training_test import CallTrainTwiceTest
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.test_export_recipe import TestExportRecipe
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.average_meter_test import TestAverageMeter
from tests.unit_tests.repvgg_unit_test import TestRepVgg
from tests.unit_tests.test_without_train_test import TestWithoutTrainTest
from tests.unit_tests.train_with_intialized_param_args_test import TrainWithInitializedObjectsTest
from tests.unit_tests.test_auto_augment import TestAutoAugment
from tests.unit_tests.ohem_loss_test import OhemLossTest
from tests.unit_tests.early_stop_test import EarlyStopTest
from tests.unit_tests.segmentation_transforms_test import SegmentationTransformsTest
from tests.unit_tests.pretrained_models_unit_test import PretrainedModelsUnitTest
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
from tests.unit_tests.post_prediction_callback_test import TestPostPredictionCallback
from tests.unit_tests.test_predict import TestModelPredict
from tests.unit_tests.test_deprecate import TestDeprecationDecorator
from tests.unit_tests.test_models_factory import DynamicModelTests
from tests.unit_tests.test_mixed_precision_cpu import TestMixedPrecisionDisabled
from tests.unit_tests.test_data_adapters import TestClassificationAdapter, TestDetectionAdapter, TestSegmentationAdapter
from tests.unit_tests.optical_flow_loss_test import OpticalFlowLossTest
from tests.unit_tests.test_optical_flow_metric import TestOpticalFlowMetric

__all__ = [
    "CrashTipTest",
    "ZeroWdForBnBiasTest",
    "SaveCkptListUnitTest",
    "TestAverageMeter",
    "TestRepVgg",
    "TestWithoutTrainTest",
    "StrictLoadEnumTest",
    "TrainWithInitializedObjectsTest",
    "TestAutoAugment",
    "OhemLossTest",
    "EarlyStopTest",
    "SegmentationTransformsTest",
    "PretrainedModelsUnitTest",
    "TestConvBnRelu",
    "FactoriesTest",
    "InitializeWithDataloadersTest",
    "TrainingParamsTest",
    "TrainOptimizerParamsOverride",
    "CallTrainTwiceTest",
    "ResumeTrainingTest",
    "CallTrainAfterTestTest",
    "ConfigInspectTest",
    "TestTransforms",
    "TestPostPredictionCallback",
    "TestModelPredict",
    "TestDeprecationDecorator",
    "TestMixedPrecisionDisabled",
    "DynamicModelTests",
    "TestExportRecipe",
    "TestClassificationAdapter",
    "TestDetectionAdapter",
    "TestSegmentationAdapter",
    "OpticalFlowLossTest",
    "TestOpticalFlowMetric",
]