Avoid annoying pandas warning.

faustomorales · Apr 20, 2023 · fea2174 · fea2174
1 parent 4c2ad0f
commit fea2174
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mira/core/scene.py b/mira/core/scene.py
@@ -612,7 +612,7 @@ def augment(
                     pd.DataFrame(transformed["keypoints"][4:], columns=["x", "y"]),
                 ],
                 axis=1,
-            ).groupby(["annIdx"])
+            ).groupby("annIdx")
         ]
         recropped = [
             ann.crop(width=image.shape[1], height=image.shape[0]) for ann in annotations