diff --git a/tasks/main.py b/tasks/main.py index 9bc38f5fd28..3c8aaab5f92 100644 --- a/tasks/main.py +++ b/tasks/main.py @@ -95,6 +95,8 @@ def get_tasks_args(parser): from orqa.evaluate_orqa import main elif args.task in ['RET-FINETUNE-NQ']: from orqa.supervised.finetune import main + elif args.task in ['EVAL-HARNESS']: + from tasks.eval_harness.evaluate import main else: raise NotImplementedError('Task {} is not implemented.'.format( args.task))