hype_kg/codes/model.py [1065:1075]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    test_triples, 
                    test_ans, 
                    test_ans_hard,
                    args.nentity, 
                    args.nrelation, 
                    'tail-batch'
                ), 
                batch_size=args.test_batch_size,
                num_workers=max(1, args.cpu_num), 
                collate_fn=TestDataset.collate_fn
            )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



hype_kg/codes/model.py [1107:1117]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    test_triples, 
                    test_ans, 
                    test_ans_hard,
                    args.nentity, 
                    args.nrelation, 
                    'tail-batch'
                ), 
                batch_size=args.test_batch_size,
                num_workers=max(1, args.cpu_num), 
                collate_fn=TestDataset.collate_fn
            )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -