Minor change to allow data_struct for qp_problems
@@ -126,6 +126,8 @@ class data_struct(object):
# Get average distance
avg_dist = np.mean(u_red - l_red) if any(u_red - l_red) else np.inf
+ return avg_dist
+
def update_data(self, seed, name, qp, results,
rho, sigma, alpha, settings):
self.n_data.append(qp.n)
