1 files changed, 5 insertions, 0 deletions

diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/autoscale/ClusterModel.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/autoscale/ClusterModel.java
index 35aafd3e0f4..1001ab83cc0 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/autoscale/ClusterModel.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/autoscale/ClusterModel.java
@@ -32,6 +32,7 @@ public class ClusterModel {
     static final double idealDiskLoad = 0.6;
 
     private final Application application;
+    private final Cluster cluster;
     /** The current nodes of this cluster, or empty if this models a new cluster not yet deployed */
     private final NodeList nodes;
     private final Clock clock;
@@ -50,6 +51,7 @@ public class ClusterModel {
                         MetricsDb metricsDb,
                         Clock clock) {
         this.application = application;
+        this.cluster = cluster;
         this.nodes = clusterNodes;
         this.clock = clock;
         this.scalingDuration = computeScalingDuration(cluster, clusterSpec);
@@ -65,6 +67,7 @@ public class ClusterModel {
                  ClusterTimeseries clusterTimeseries,
                  ClusterNodesTimeseries nodeTimeseries) {
         this.application = application;
+        this.cluster = cluster;
         this.nodes = null;
         this.clock = clock;
 
@@ -73,6 +76,8 @@ public class ClusterModel {
         this.nodeTimeseries = nodeTimeseries;
     }
 
+    public Cluster cluster() { return cluster; }
+
     /** Returns the predicted duration of a rescaling of this cluster */
     public Duration scalingDuration() { return scalingDuration; }