Minor fix to image listing

OZGrav · Nov 7, 2023 · 405a4cf · 405a4cf
1 parent 6cb1b38
commit 405a4cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/psrdb/tables/pipeline_image.py b/psrdb/tables/pipeline_image.py
@@ -10,12 +10,12 @@ def __init__(self, client, token):
         GraphQLTable.__init__(self, client, token)
         self.record_name = "pipeline_image"
 
-        self.field_names = ["id", "image", "imageType", "rank", "processing {id}"]
+        self.field_names = ["id", "image", "imageType", "resolution", "cleaned", "pipelineRun {id}"]
 
-    def list(self, id=None, processing_id=None):
-        """Return a list of records matching the id and/or the processing id."""
+    def list(self, id=None, pipeline_run_id=None):
+        """Return a list of records matching the id and/or the pipelineRun id."""
         filters = [
-            {"field": "processing", "value": processing_id},
+            {"field": "processing", "value": pipeline_run_id},
         ]
         return GraphQLTable.list_graphql(self, self.table_name, filters, [], self.field_names)
 

diff --git a/psrdb/tables/pulsar_fold_result.py b/psrdb/tables/pulsar_fold_result.py
@@ -13,6 +13,7 @@ def __init__(self, client, token):
             "observation { utcStart }",
             "observation { band }",
             "observation { duration }",
+            "pipelineRun { id }",
             "pipelineRun { dm }",
             "pipelineRun { dmErr }",
             "pipelineRun { dmEpoch }",