diff --git a/chi/lease.py b/chi/lease.py index 432864d..bc62f51 100644 --- a/chi/lease.py +++ b/chi/lease.py @@ -308,6 +308,10 @@ def add_node_reservation( reservation_list=self.node_reservations, node_name=node.name ) else: + if not amount or not (node_type or node_name): + raise CHIValueError( + "You must specify amount and either node_type or node_name" + ) add_node_reservation( reservation_list=self.node_reservations, count=amount,