From ef17ffa84c553c3c7d4456dd9a51d8b020172468 Mon Sep 17 00:00:00 2001
From: Ben Elliston <bje@air.net.au>
Date: Fri, 12 Jan 2024 09:44:35 +1100
Subject: [PATCH] * nemo/generators.py (PumpedHydroPump.__init__): Allow
 adjustment of   capacity in the range between zero and the initial capacity.

---
 nemo/generators.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nemo/generators.py b/nemo/generators.py
index 917ab471..8b9f1e8b 100644
--- a/nemo/generators.py
+++ b/nemo/generators.py
@@ -438,6 +438,8 @@ def __init__(self, polygon, capacity, reservoirs, rte=0.8, label=None):
             raise TypeError
         Storage.__init__(self)
         Generator.__init__(self, polygon, capacity, label)
+        # capacity is in MW, but build limit is in GW
+        self.setters = [(self.set_capacity, 0, capacity / 1000.)]
         self.reservoirs = reservoirs
         self.rte = rte