restore to original main.py

tasks/main.py

@@ -84,28 +84,6 @@ def get_tasks_args(parser):
  help='Av.rank validation: how many other negatives to'
  ' take from each question pool')
 
- # parameters for the knowledgeable dialogue generation
- group.add_argument("--sample-input-file", type=str, default=None,
- help='Get input from file instead of interactive mode, '
- 'each line is an input.')
- group.add_argument("--sample-output-file", type=str, default=None,
- help='Output file got from --sample-input-file')
- group.add_argument('--prompt-file', type=str, default=None,
- help='prompting file')
- group.add_argument('--prompt-type', type=str, default=None,
- help='prompt type (knowledge or response)')
- group.add_argument('--num-prompt-examples', type=int, default=10,
- help='number of prompt examples')
- group.add_argument('--guess-file', type=str, default=None,
- help='datapath for generated sentences')
- group.add_argument('--answer-file', type=str, default=None,
- help='datapath for golden sentences')
- group.add_argument('--out-seq-length', type=int, default=100,
- help='output sequence length')
- group.add_argument('--api-prompt', default=False, action="store_true",
- help='setup model api for prompting')
- group.add_argument('--megatron-api-url', type=str, default=None,
- help='url of the megatron api')
 
  return parser
 
@@ -130,10 +108,6 @@ 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 == 'KNWL-DIALO-PROMPT':
- from knwl_dialo.prompt import main
- elif args.task == 'KNWL-DIALO-EVAL-F1':
- from knwl_dialo.evaluate import main
  else:
  raise NotImplementedError('Task {} is not implemented.'.format(
  args.task))