Update kaleido and plotly versions (#163)

* Update versions

* Remove logger and simplify output names

* Add test

pyodi/apps/ground_truth.py

@@ -48,15 +48,12 @@
 from pathlib import Path
 from typing import Optional, Tuple
 
-from loguru import logger
-
 from pyodi.core.boxes import add_centroids
 from pyodi.core.utils import coco_ground_truth_to_df
 from pyodi.plots.boxes import get_centroids_heatmap, plot_heatmap
 from pyodi.plots.common import plot_scatter_with_histograms
 
 
-@logger.catch
 def ground_truth(
     ground_truth_file: str,
     show: bool = True,
@@ -87,7 +84,7 @@ def ground_truth(
         df_images,
         x="img_width",
         y="img_height",
-        title=f"{Path(ground_truth_file).stem}: Image Shapes",
+        title="Image_Shapes",
         show=show,
         output=output,
         output_size=output_size,
@@ -108,7 +105,7 @@ def ground_truth(
         df_annotations,
         x="absolute_width",
         y="absolute_height",
-        title=f"{Path(ground_truth_file).stem}: Bounding Box Shapes",
+        title="Bounding_Box_Shapes",
         show=show,
         output=output,
         output_size=output_size,
@@ -120,7 +117,7 @@ def ground_truth(
 
     plot_heatmap(
         get_centroids_heatmap(df_annotations),
-        title=f"{Path(ground_truth_file).stem}: Bounding Box Centers",
+        title="Bounding_Box_Centers",
         show=show,
         output=output,
         output_size=output_size,

setup.cfg

@@ -17,9 +17,9 @@ install_requires =
   numba==0.52.0
   pandas==1.2.1
   pillow==8.1.0
-  plotly==4.14.3
+  plotly==5.2.2
   pycocotools==2.0.2
-  kaleido==v0.1.0
+  kaleido==v0.2.1
   scikit-learn==0.24.1
   fire==0.4.0
 

tests/apps/test_ground_truth.py

@@ -0,0 +1,27 @@
+import json
+from pathlib import Path
+
+from pyodi.apps.ground_truth import ground_truth
+
+
+def test_ground_truth_saves_output_to_files(tmpdir):
+    output = tmpdir.mkdir("results")
+
+    categories = [{"id": 1, "name": "drone"}]
+    images = [{"id": 0, "file_name": "image.jpg", "height": 10, "width": 10}]
+    annotations = [
+        {"image_id": 0, "category_id": 1, "id": 0, "bbox": [0, 0, 5, 5], "area": 25}
+    ]
+    coco_data = dict(
+        images=images,
+        annotations=annotations,
+        categories=categories,
+        info={},
+        licenses={},
+    )
+    with open(tmpdir / "data.json", "w") as f:
+        json.dump(coco_data, f)
+
+    ground_truth(tmpdir / "data.json", show=False, output=output)
+
+    assert len(list(Path(output / "data").iterdir())) == 3