device
- Handle the new `device` parameter in PySpark. - Deprecate the old `use_gpu` parameter.
- Move test files. - Run spark and dask separately to prevent conflicts. - Gather common code into the testing module.