The following script with n_workers=1 runs successfully. With n_workers=2 then it gives an error: "AssertError:bu