[QUESTION]: Error with Import ray.train

[aashuph16221]

Hello,

I am currently testing MolPAL and encountered an issue when running it with the following configuration:

[model] model = mpn conf-method = mve

Despite updating the trainer, I continue to receive the following error:

base) [aashish@pharmcluster3 fingerprint]$ tail -f slurm-3331.out explorer = Explorer(path, **params) File "/shared/src/molpal/molpal/explorer.py", line 147, in init self.model = models.model(input_size=len(self.featurizer), **kwargs) File "/shared/src/molpal/molpal/models/init.py", line 27, in model return mpn(**kwargs) File "/shared/src/molpal/molpal/models/init.py", line 55, in mpn from molpal.models.mpnmodels import MPNModel, MPNDropoutModel, MPNTwoOutputModel File "/shared/src/molpal/molpal/models/mpnmodels.py", line 16, in from ray.train import Trainer as RayTrainer ImportError: cannot import name 'Trainer' from 'ray.train' (/shared/apps/anaconda3/2024.02/envs/molpal/lib/python3.8/site-packages/ray/train/init.py)

It seems there may be an issue with the import from ray.train. Could you kindly advise if this is related to a version conflict, a missing dependency, or a potential bug? Any guidance on resolving this issue would be greatly appreciated.

Thank you for your time and support!

Regards,

Aashish