MAINT: raise NotImplementedError -> return NotImplemented