style: add new line at the end of the file

nilmtk · Oct 7, 2020 · c8903ad · c8903ad
2 parents 04b64d7 + 8f5768f
commit c8903ad
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nilmtk_contrib/disaggregate/afhmm_sac.py b/nilmtk_contrib/disaggregate/afhmm_sac.py
@@ -209,7 +209,11 @@ def disaggregate_thread(self, test_mains,index,d):
                 u = time.time()
                 prob = cvx.Problem(expression, constraints)
                 prob.solve(solver=cvx.SCS,verbose=False, warm_start=True)
-                s_ = [i.value for i in cvx_state_vectors]
+                s_ = [
+                    np.zeros((len(test_mains), self.default_num_states)) if i.value is None
+                    else i.value
+                    for i in cvx_state_vectors
+                ]
 
         prediction_dict = {}
         for appliance_id in range(self.num_appliances):