Fix test which required accelerate, apply style

[kylesayrs]

Purpose

• Fix test which was failing without accelerate installed

Changes

• Reuse helper function to avoid passing device_map
• Apply style

[dsikka]

Which test was failing, the mse or apply test?
Was it just skipped in the ci?

[kylesayrs]

@dsikka Test fails if accelerate is not installed. Small change but nicer experience for users who do the following

pip install compressed-tensors
pytest tests