Optimize dataset usage
Remove using torch tensors from the path context dataset. Now, it returns lists of ints. This should speed up calculations.
Remove using torch tensors from the path context dataset. Now, it returns lists of ints. This should speed up calculations.