Small bug in train-latent-probe.ipynb

[benbenwsh]

Small bug in load_dataset function. After changing the directory using os.chdir, you forgot to change the directory back to the original, leading to the loop

for path in ds_paths:
    Ds.append(Dataset(load_dataset(path)))

to fail

[JThh]

Thank you for catching this bug. Will issue a fix on this!