updated BigQuery output

Author: Newake

broker/cloud_run/lsst/scone/main.py

@@ -95,15 +95,13 @@ def run():
 
     # publish
     outpt_dict = {
-        "diaObjectId": alert_lite.dict["alert_lite"]["diaObject"]["diaObjectId"],
-        "diaSourceId": alert_lite.dict["alert_lite"]["diaSource"]["diaSourceId"],
         "prob": scone_classification,
         "predicted_class": round(scone_classification),
     }
 
     TOPIC.publish(
         pittgoogle.Alert.from_dict(
-            payload={"alert_lite": alert_lite.dict, "SCONE": outpt_dict},
+            payload={"alert_lite": alert_lite.dict['alert_lite'], "SCONE": outpt_dict},
             attributes={
                 **alert_lite.attributes,
                 "pg_scone_class": outpt_dict["predicted_class"],

broker/setup_broker/lsst/templates/bq_lsst_SCONE_schema.json

@@ -0,0 +1,58 @@
+[
+    {
+      "description": "Unique identifiers from the semantically compressed alert packet.",
+      "fields":[
+          {
+            "description": "diaSource",
+            "fields": [
+                {
+                  "description": "Unique identifier of this diaSource.",
+                  "mode": "REQUIRED",
+                  "name": "diaSourceId",
+                  "type": "INTEGER"
+                }
+            ],
+            "mode": "REQUIRED",
+            "name": "diaSource",
+            "type": "RECORD"
+          },
+          {
+            "description": "diaObject",
+            "fields": [
+                {
+                  "description": "Unique identifier of this diaObject.",
+                  "mode": "REQUIRED",
+                  "name": "diaObjectId",
+                  "type": "INTEGER"
+                }
+            ],
+            "mode": "REQUIRED",
+            "name": "diaObject",
+            "type": "RECORD"
+          }
+      ],
+      "mode": "REQUIRED",
+      "name": "alert_lite",
+      "type": "RECORD"
+    },
+    {
+      "description": "Classification results using SCONE (Helen Qu et al. 2021).",
+      "fields": [
+          {
+            "description": "probability object is a type Ia supernova",
+            "mode": "NULLABLE",
+            "name": "prob",
+            "type": "FLOAT"
+          },
+          {
+            "description": "predicted class; 1 = Ia, 0 = non-Ia",
+            "mode": "NULLABLE",
+            "name": "predicted_class",
+            "type": "INTEGER"
+          }
+      ],
+      "mode": "REQUIRED",
+      "name": "SCONE",
+      "type": "RECORD"
+    }
+]