Python TypeError: 'Tensor' object is not callable when sorting dictionary