Fix readthedocs build process.

faustomorales · Aug 18, 2022 · cd0db51 · cd0db51
1 parent f7e4bb2
commit cd0db51
Show file tree

Hide file tree

Showing 7 changed files with 21 additions and 10 deletions.
diff --git a/.readthedocs.yaml b/.readthedocs.yaml
@@ -12,9 +12,16 @@ formats: []
 
 # Installs the package and the docs requirements.
 python:
-  version: 3.7
   install:
     - requirements: docs/requirements.txt
     - method: pip
       path: .
   system_packages: true
+
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.8"
+  jobs:
+    pre_install:
+      - make patch-thirdparty
diff --git a/docs/conf.py b/docs/conf.py
@@ -73,7 +73,7 @@
 #
 # This is also used if you do content translation via gettext catalogs.
 # Usually you set "language" from the command line for these cases.
-language = None
+language = "en"
 
 # List of patterns, relative to source directory, that match files and
 # directories to ignore when looking for source files.

diff --git a/docs/detectors/efficientdet.rst b/docs/detectors/efficientdet.rst
diff --git a/docs/detectors/index.rst b/docs/detectors/index.rst
@@ -10,4 +10,4 @@ There are currently four implemented architectures.
 
 .. toctree::
    fasterrcnn
-   efficientdet
+   retinanet
diff --git a/docs/detectors/retinanet.rst b/docs/detectors/retinanet.rst
@@ -0,0 +1,4 @@
+EfficientDet
+************
+
+.. autoclass:: mira.detectors.RetinaNet
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -11,4 +11,7 @@ torchvision
 albumentations
 mistune<2
 regex
-ftfy
+ftfy
+timm
+pretrainedmodels
+efficientnet_pytorch
diff --git a/mira/metrics.py b/mira/metrics.py
@@ -175,12 +175,13 @@ def crop_error_examples(
     iou_threshold=0.1,
 ) -> typing.List[typing.Dict[str, typing.List[Annotation]]]:
     """Get crops of true positives, false negatives, and false positives.
+
     Args:
         true_collection: A collection of the ground truth scenes.
         pred_collection: A collection of the predicted scenes.
-        threshold: The score threshold for selecting annotations from predicted
-            scenes.
+        threshold: The score threshold for selecting annotations from predicted scenes.
         iou_threhsold: The IoU threshold for counting a box as a true positive.
+
     Returns:
         A list of dicts with "tps", "fps", and "fns"
         with the same length of the input collections. The values in each dict