InternalError: Check failed: type_code_ == kDLDevice (8 vs. 6) : expected DLDevice but got Object

Yes, this works. Thank you