replace OP_NAME

modelscope · Nov 28, 2023 · e2c7970 · e2c7970
1 parent 037baec
commit e2c7970
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data_juicer/ops/filter/face_area_filter.py b/data_juicer/ops/filter/face_area_filter.py
@@ -34,8 +34,8 @@ def get_classifier(self):
         return cv2.CascadeClassifier(cv2.data.haarcascades + self.file_path)
 
 
-@OPERATORS.register_module('face_area_filter')
-@LOADED_IMAGES.register_module('face_area_filter')
+@OPERATORS.register_module(OP_NAME)
+@LOADED_IMAGES.register_module(OP_NAME)
 class FaceAreaFilter(Filter):
     """Filter to keep samples with face area ratio within a specific range.
     """