Fix: raise explicit memory error when dataset allocation fails

Fix: raise explicit memory error when dataset allocation fails #303