lmap: Better determine like device using un-deprecated method (!907) · Merge requests · ift / NIFTy

Gordian Edenhofer requested to merge undeprecate_lmap into NIFTy_8 Dec 24, 2023

JAX deprecated ndarray.device in favor of ndarray.devices (with an s) returning a list of devices instead of a single device.

Edited Jan 01, 2024 by Gordian Edenhofer

lmap: Better determine like device using un-deprecated method