#!/usr/bin/env python3
# args: base out stage1_cache out_root ds exp views sched_csv batch_size single(0/1)
import sys
base,out,stage1_cache,out_root,ds,exp,views,sched,bs,single = sys.argv[1:11]
txt=open(base).read()
idx=txt.index('\nval_appforce:'); head=txt[:idx]
sched_list='['+', '.join(sched.split(','))+']'
single_yaml = 'true' if str(single) in ('1','true','True') else 'false'
block=f"""
val_appforce:
  enabled: true
  every: 5000
  at_step0: false
  views: {views}
  schedules: {sched_list}
  seed: 42
  limit: 0
  batch_size: {int(bs)}
  single: {single_yaml}
  decode_procs: 1
  ref_threads: 8
  stage1_cache: {stage1_cache}
  out_root: {out_root}
  ref_python: /lp-dev/jonghoon/mv-mesh/envs/mv-sam3d/bin/python
  eval_dataset_flag: toys4k
  datasets:
    - name: {ds}
      exp: {exp}
"""
open(out,'w').write(head+block); print("wrote",out,"batch_size",bs,"single",single_yaml)