In case anyone stumbles across this whilst using Assembly Definitions...
You'll need to explicitly add a reference to `UnityEngine.InputSystem` to the relevant assembly definition