fix: Fix lint and tests.

tests/docs_test.py

@@ -26,7 +26,7 @@
 #         exec_timeout=50, diff_ignore=ignores, diff_replace=replace
 #     )
 #     fixture.diff_color_words = False
-#     file = "docs/source/tutorials/3D_visualization.ipynb"
+#     file = "docs/source/user_guide/3D_visualization.ipynb"
 #     fixture.check(file)
 
 
@@ -52,48 +52,29 @@ def test_vis() -> None:
         exec_timeout=50, diff_ignore=ignores, diff_replace=replace
     )
     fixture.diff_color_words = False
-    file = "docs/source/tutorials/visualization.ipynb"
+    file = "docs/source/user_guide/visualization.ipynb"
     fixture.check(file)
 
 
 def test_get_started() -> None:
     """Test get started notebooks."""
     ignores = (
         "/cells/*/metadata",
-        "/cells/1/outputs/",  # Suppress downloading checkpoint output
+        "/cells/3/outputs/",  # Suppress downloading checkpoint output
         "/metadata/widgets",
         "/cells/5/outputs/",  # Suppress downloading checkpoint output
+        "/cells/9/outputs/",  # Suppress downloading checkpoint output
+        "/cells/11/outputs/",  # Suppress downloading checkpoint output
         "/cells/*/execution_count",
         "/cells/*/outputs/*/data/image",
         "/metadata/language_info/version",
     )
     replace = (("/cells/*/outputs", "\\[Open3D INFO\\] [^\\n]+ *\\n?", ""),)
 
-    fixture = (
-        NBRegressionFixture(  # Higher timeout for downloading checkpoints
-            exec_timeout=300, diff_ignore=ignores, diff_replace=replace
-        )
-    )
-    fixture.diff_color_words = False
-    file = "docs/source/tutorials/getting_started.ipynb"
-    fixture.check(file)
-
-
-def test_cli() -> None:
-    """Test the cli notebook."""
-    # Lets for now ignore the output of the training cells. Cuda/Python version
-    # on the server are different and the training is slightly non-det.
-    ignores = (
-        "/cells/2/outputs/",
-        "/cells/4/outputs/",
-        "/cells/*/metadata",
-        "/metadata/language_info/version",
-    )
-    fixture = (
-        NBRegressionFixture(  # Higher timeout for downloading checkpoints
-            exec_timeout=300, diff_ignore=ignores
-        )
+    # Higher timeout for downloading checkpoints
+    fixture = NBRegressionFixture(
+        exec_timeout=300, diff_ignore=ignores, diff_replace=replace
     )
     fixture.diff_color_words = False
-    file = "docs/source/tutorials/cli_interface.ipynb"
+    file = "docs/source/user_guide/getting_started.ipynb"
     fixture.check(file)

tests/model/detect/mask_rcnn_test.py

@@ -98,7 +98,9 @@ def test_train(self):
         rcnn_box_encoder, _ = get_default_rcnn_box_codec()
         rcnn_loss = RCNNLoss(rcnn_box_encoder)
 
-        mask_loss = SampledMaskLoss(positive_mask_sampler, MaskRCNNHeadLoss())
+        mask_loss = SampledMaskLoss(
+            positive_mask_sampler, MaskRCNNHeadLoss(num_classes=80)
+        )
 
         mask_rcnn_loss = LossModule(
             [

vis4d/op/detect/rcnn.py

@@ -197,6 +197,7 @@ def __init__(
         max_per_img: int = 100,
     ) -> None:
         """Creates an instance of the class.
+
         Args:
             box_decoder (DeltaXYWHBBoxDecoder, optional): Decodes regression
                 parameters to detected boxes. Defaults to None. If None, it
@@ -310,6 +311,7 @@ def __init__(
         loss_bbox: TorchLossFunc = l1_loss,
     ) -> None:
         """Creates an instance of the class.
+
         Args:
             box_encoder (DeltaXYWHBBoxEncoder): Decodes box regression
                 parameters into detected boxes.