o *j@s4ddlZddlmZddlZddlmZddZdS)N) Namespace)ConstructBlockStrategycCst}d|_d|_t|dddd}tdgd}tdD]%}|tjgdtdd d gt }|D]\}}|||d 7<q3qt |dS) N rg?)Z bert_ratioZmax_seq_lengthi') r r)Zdtyper) rZ seq_lengthZ eod_tokenrnparrayrangeZsample_span_in_documentintrandomRandomprint)argsZstrategycounts_spansstartendrk/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/nlp/mglm/test/test_block.pymains    r)rargparsernumpyrZ blocklm_utilsrrrrrrs