in pretrain_dataset.py, you used client from petrel_client, but it's not a package covered in requirements.txt.
And directly installing by pip install petrel_client doesn't work.
Where can I find and install this package?