Hi Felix,
ah, yes, this extra functionality has been written assuming the presence of hydro units.
You'd need to rewrite the function such that code passages handling the storage are skipped, e.g. by testing wether n.storage_units is empty:
if not n.storage_units.empty:
# add storage terms to constraint
Best wishes.
Fabian N