Updated datamodules.py to work for any OS

MolecularAI / Chemformer

Apache License 2.0

211 stars 36 forks source link

Closed salman-moh closed 2 years ago

salman-moh commented 2 years ago

Helps resolve the following error when predicting on Windows/OSX systems.

self._num_workers = len(os.sched_getaffinity(0))
AttributeError: module 'os' has no attribute 'sched_getaffinity'

salman-moh commented 2 years ago

Need to change os.sched_getaffinity(0) in other files as well.