Add check for amount when using node_type

ChameleonCloud · Oct 30, 2024 · 9fdd021 · 9fdd021
1 parent 972aa22
commit 9fdd021
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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,