o h<@snddlZddlmZddlmZddlmZdZddZdd Z e d kr5eZ e d Z d Z e e e d dSdS)N)Tree) read_treebank)get_default_paths)traindevtestcCsN|jdr td|}|jdrtd|}n |jdr"td|}td|S)NNNNPNEPNXYVROOTROOT)label startswithr)treerk/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/utils/datasets/constituency/convert_spmrl.pyadd_root s       rc CstD]O}tj|||d}t|td}tj|d||f}t|ddd}|D]}|t||dq)Wdn1sBwYt dt ||fqdS) Nz.German.gold.ptb) tree_callbackz %s_%s.mrgwzutf-8)encoding zWrote %d trees to %s) SHARDSospathjoinrropenwritestrprintlen) input_directoryoutput_directory short_nameshard tree_filenametreesoutput_filenamefoutrrrr convert_spmrls  r*__main__CONSTITUENCY_DATA_DIRzFextern_data/constituency/spmrl/SPMRL_SHARED_2014/GERMAN_SPMRL/gold/ptbde_spmrl)r%stanza.models.constituency.parse_treer&stanza.models.constituency.tree_readerrstanza.utils.default_pathsrrrr*__name__pathsr#r"rrrrs