Update PanopticQuality.py

PanopticQuality.py

@@ -87,6 +87,7 @@ class PQMetric(evaluate.Metric):
                                         (6**2, 12**2),
                                         (12**2, 1e5**2)],
         method: Literal["iou", "hungarian"] = "iou",
+        device: str = None,
         class_agnostic: bool = False,
         **kwargs
     ):
@@ -140,6 +141,7 @@ class PQMetric(evaluate.Metric):
         self.stuff =  stuff if stuff is not None else DEFAULT_STUFF
         self.per_class = per_class
         self.split_sq_rq = split_sq_rq
+        self.device = device
         self.pq_metric = PanopticQuality(
             things=set([self.label2id[label] for label in self.label2id.keys() if label not in self.stuff]), 
             stuffs=set([self.label2id[label] for label in self.label2id.keys() if label in self.stuff]),