diff --git a/mindarmour/reliability/__init__.py b/mindarmour/reliability/__init__.py
index 9093e46..84b7234 100644
--- a/mindarmour/reliability/__init__.py
+++ b/mindarmour/reliability/__init__.py
@@ -12,9 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """
-Reliability methods of MindArmour
+Reliability methods of MindArmour.
 """
 
 from .model_fault_injection.fault_injection import FaultInjector
+from .concept_drift.concept_drift_check_time_series import ConceptDriftCheckTimeSeries
+from .concept_drift.concept_drift_check_images import OodDetector
+from .concept_drift.concept_drift_check_images import OodDetectorFeatureCluster
 
-__all__ = ['FaultInjector']
+__all__ = ['FaultInjector',
+           'ConceptDriftCheckTimeSeries',
+           'OodDetector',
+           'OodDetectorFeatureCluster']
diff --git a/mindarmour/reliability/concept_drift/concept_drift_check_images.py b/mindarmour/reliability/concept_drift/concept_drift_check_images.py
index 10c3daa..361a311 100644
--- a/mindarmour/reliability/concept_drift/concept_drift_check_images.py
+++ b/mindarmour/reliability/concept_drift/concept_drift_check_images.py
@@ -178,6 +178,7 @@ class OodDetectorFeatureCluster(OodDetector):
     def ood_predict(self, threshold, ds_test):
         """
         The out-of-distribution detection.
+
         Args:
             threshold (float): the threshold to judge ood data. One can set value by experience
                 or use function get_optimal_threshold.