Fix prediction results for unlabeled data

Author: bnaul

cesium_app/models.py

@@ -208,7 +208,10 @@ def is_owned_by(self, username):
     def format_pred_data(fset, data):
         fset.columns = fset.columns.droplevel('channel')
         fset.index = fset.index.astype(str)  # can't use ints as JSON keys
-        labels = pd.Series(data.get('labels'), index=fset.index)
+
+        labels = pd.Series(data['labels'] if len(data.get('labels', [])) > 0
+                           else None, index=fset.index)
+
         if len(data.get('pred_probs', [])) > 0:
             preds = pd.DataFrame(data.get('pred_probs', []),
                                  index=fset.index).to_dict(orient='index')

cesium_app/tests/frontend/test_predict.py

@@ -137,6 +137,25 @@ def test_pred_results_table_regr(driver):
             raise
 
 
+def test_add_prediction_unlabeled(driver):
+    driver.get('/')
+    with create_test_project() as p,\
+         create_test_dataset(p, label_type=None) as ds,\
+         create_test_featureset(p) as fs, create_test_model(fs) as m:
+        _add_prediction(p.id, driver)
+        _click_prediction_row(p.id, driver)
+        try:
+            rows = _grab_pred_results_table_rows(driver, 'Mira')
+            for row in rows:
+                probs = [float(v.text)
+                         for v in row.find_elements_by_tag_name('td')[2::2]]
+                assert sorted(probs, reverse=True) == probs
+            driver.find_element_by_xpath("//th[contains(text(),'Time Series')]")
+        except:
+            driver.save_screenshot("/tmp/pred_click_tr_fail.png")
+            raise
+
+
 def test_delete_prediction(driver):
     driver.get('/')
     with create_test_project() as p, create_test_dataset(p) as ds,\