o ߥi@s4ddlZddlmZddlZddlmZddZdS)N) Namespace)ConstructBlockStrategycCst}d|_d|_t|dddd}tdgd}tdD]%}|tjgdtdd d gt }|D]\}}|||d 7<q3qt |dS) N rg?) bert_ratiomax_seq_lengthi') r r)dtyper) r seq_length eod_tokenrnparrayrangesample_span_in_documentintrandomRandomprint)argsstrategycounts_spansstartendr"^/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/nlp/mglm/test/test_block.pymains    r$)rargparsernumpyr blocklm_utilsrr$r"r"r"r#s