refactor(reward,dataset): add __all__ for public API exports

rchardx · rchardx · commit fd268da595dc · 2026-03-07T01:08:12.000+08:00
- areal/reward/__init__.py: Add __all__ for VALID_REWARD_FN,  get_custom_reward_fn, MathVerifyWorker, get_math_verify_worker
- areal/dataset/__init__.py: Add __all__ for VALID_DATASETS
  get_custom_dataset
diff --git a/areal/dataset/__init__.py b/areal/dataset/__init__.py
@@ -151,3 +151,9 @@ def get_custom_dataset(
         processor=processor,
         **kwargs,
     )
+
+
+__all__ = [
+    "VALID_DATASETS",
+    "get_custom_dataset",
+]
diff --git a/areal/reward/__init__.py b/areal/reward/__init__.py
@@ -75,3 +75,11 @@ def get_math_verify_worker() -> MathVerifyWorker:
     if _MATH_VERIFY_WORKER is None:
         _MATH_VERIFY_WORKER = MathVerifyWorker()
     return _MATH_VERIFY_WORKER
+
+
+__all__ = [
+    "VALID_REWARD_FN",
+    "get_custom_reward_fn",
+    "MathVerifyWorker",
+    "get_math_verify_worker",
+]
