From 62f55987f17c33ebd1fe7349bfc0771c71d75000 Mon Sep 17 00:00:00 2001
From: Fengzhe Zhou <zhoufengzhe07@gmail.com>
Date: Thu, 11 Jul 2024 19:59:35 +0800
Subject: [PATCH] force register (#1311)

---
 opencompass/tasks/openicl_attack.py | 2 +-
 opencompass/tasks/openicl_eval.py   | 2 +-
 opencompass/tasks/openicl_infer.py  | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/opencompass/tasks/openicl_attack.py b/opencompass/tasks/openicl_attack.py
index 09190dc7..773fce74 100644
--- a/opencompass/tasks/openicl_attack.py
+++ b/opencompass/tasks/openicl_attack.py
@@ -15,7 +15,7 @@ from opencompass.utils import (build_dataset_from_cfg, build_model_from_cfg,
                                task_abbr_from_cfg)
 
 
-@TASKS.register_module(force=(__name__ == '__main__'))  # A hack for script run
+@TASKS.register_module()
 class OpenICLAttackTask(BaseTask):
     """OpenICL Inference Task.
 
diff --git a/opencompass/tasks/openicl_eval.py b/opencompass/tasks/openicl_eval.py
index 8d2494d8..1fd9e7ed 100644
--- a/opencompass/tasks/openicl_eval.py
+++ b/opencompass/tasks/openicl_eval.py
@@ -55,7 +55,7 @@ def extract_role_pred(s: str, begin_str: Optional[str],
     return s[start:end]
 
 
-@TASKS.register_module(force=(__name__ == '__main__'))  # A hack for script run
+@TASKS.register_module()
 class OpenICLEvalTask(BaseTask):
     """OpenICL Evaluation Task.
 
diff --git a/opencompass/tasks/openicl_infer.py b/opencompass/tasks/openicl_infer.py
index a4a83190..0157cbcf 100644
--- a/opencompass/tasks/openicl_infer.py
+++ b/opencompass/tasks/openicl_infer.py
@@ -17,7 +17,7 @@ from opencompass.utils import (build_dataset_from_cfg, build_model_from_cfg,
                                model_abbr_from_cfg, task_abbr_from_cfg)
 
 
-@TASKS.register_module(force=(__name__ == '__main__'))  # A hack for script run
+@TASKS.register_module()
 class OpenICLInferTask(BaseTask):
     """OpenICL Inference Task.