o wi?@sddlmZddlZddlmZmZmZedddgZeeeeeddZ edeeeedZ edeeeed d dZ eeeeeeeeed d dZ edddZ dS) ) namedtupleN) BATCH_SIZE NUM_BATCHES NUM_CLASSESInputprobslogits)rr d) collectionsrtorchpl_utilsrrrrrand ONLY_PROBS ONLY_LOGITS1ONLY_LOGITS100PROBS_AND_LOGITSNO_PROBS_NO_LOGITSrrg/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/tests/collections/common/perplexity_inputs.pys