minor update on power scheduling eval

tasks/power_scheduling/eval.py

@@ -17,22 +17,22 @@ def task_loss(Y_sched, Y_actual, params):
 def load_indexed_json(path):
     with open(path) as f:
         data = json.load(f)
-    rows = [data[k] for k in sorted(data.keys(), key=int)]
-    return np.array(rows, dtype=np.float64)
+    keys = sorted(data.keys(), key=int)
+    return np.array([data[k] for k in keys], dtype=np.float64), keys
 
 
 def load_predicted_json(path):
     with open(path) as f:
         data = json.load(f)
     keys = sorted(data.keys(), key=int)
-    Z = np.array([data[k] for k in keys], dtype=np.float64)
-    return Z
+    return np.array([data[k] for k in keys], dtype=np.float64), keys
 
 
 def evaluate(pred_path, actual_path, params=PARAMS):
-    Z_pred = load_predicted_json(pred_path)
-    Y_actual = load_indexed_json(actual_path)
-    
+    Z_pred, p_inst = load_predicted_json(pred_path)
+    Y_actual, a_inst = load_indexed_json(actual_path)
+
+    assert p_inst == a_inst, "instance ids mismatch between pred and actual"
     assert Z_pred.shape == Y_actual.shape, \
         f"z shape mismatch: pred {Z_pred.shape} vs actual {Y_actual.shape}"