o }oi @sddlZddlZddlmZejdejgdgdgdgejddd fejgd gd gejddd fejd ejdd dfejgdgdgdgejdddfgddZ dS)N)count_tail_paddingzlabels,expected,ignore_label)rrr)rr)rrr)dtyperr)rrr)rrr )rr ) r)rrrr)r rrr rcCs t||d}t||ksJdS)N) ignore_label)rint)labelsexpectedroutrO/home/ubuntu/.local/lib/python3.10/site-packages/tests/automodel/_test_utils.pytest_count_tail_paddings 2r) pytesttorch:nemo.collections.llm.gpt.model.hf_auto_model_for_causal_lmrmark parametrizetensorlongzerosrrrrrsR   1