Merge pull request #146 from Unity-Technologies/0.2.x-dependency-update

update dependencies to newer versions for sprawl compatibility
Unity-Technologies · Feb 24, 2021 · 022fd4c · 022fd4c
2 parents d1e09d2 + 2bd1fe4
commit 022fd4c
Show file tree

Hide file tree

Showing 4 changed files with 1,202 additions and 962 deletions.
diff --git a/datasetinsights/evaluation_metrics/average_precision_2d.py b/datasetinsights/evaluation_metrics/average_precision_2d.py
@@ -269,7 +269,7 @@ class MeanAveragePrecisionAverageOverIOU(EvaluationMetric):
     TYPE = "scalar"
 
     IOU_THRESHOULDS = np.linspace(
-        0.5, 0.95, np.round((0.95 - 0.5) / 0.05) + 1, endpoint=True
+        0.5, 0.95, int(np.round((0.95 - 0.5) / 0.05)) + 1, endpoint=True
     )
 
     def __init__(self):

diff --git a/datasetinsights/evaluation_metrics/average_recall_2d.py b/datasetinsights/evaluation_metrics/average_recall_2d.py
@@ -111,7 +111,7 @@ class MeanAverageRecallAverageOverIOU(EvaluationMetric):
     TYPE = "scalar"
 
     IOU_THRESHOULDS = np.linspace(
-        0.5, 0.95, np.round((0.95 - 0.5) / 0.05) + 1, endpoint=True
+        0.5, 0.95, int(np.round((0.95 - 0.5) / 0.05)) + 1, endpoint=True
     )
 
     def __init__(self):