Update PanopticQuality.py

PanopticQuality.py

@@ -141,13 +141,14 @@ 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
+        print("HF: 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]),
             return_per_class=per_class,
             return_sq_and_rq=split_sq_rq,
             areas=area_rng,
+            device=device
             #method=method
         )
         self.cont_to_cat = {label:key for key, label in self.pq_metric.metric.cat_id_to_continuous_id.items()}