Fix test

Author: qubvel

tests/models/base.py

@@ -99,6 +99,9 @@ def test_in_channels_and_depth_and_out_classes(
         if self.model_type in ["unet", "unetplusplus", "manet"]:
             kwargs = {"decoder_channels": self.decoder_channels[:depth]}
 
+        if self.model_type == "dpt":
+            kwargs = {"decoder_intermediate_channels": self.decoder_channels[:depth]}
+
         model = (
             smp.create_model(
                 arch=self.model_type,

tests/models/test_dpt.py

@@ -1,4 +1,5 @@
 import pytest
+import inspect
 import torch
 import segmentation_models_pytorch as smp
 
@@ -22,6 +23,11 @@ class TestDPTModel(base.BaseModelTester):
 
     compile_dynamic = False
 
+    @property
+    def decoder_channels(self):
+        signature = inspect.signature(self.model_class)
+        return signature.parameters["decoder_intermediate_channels"].default
+
     @property
     def hub_checkpoint(self):
         return "smp-test-models/dpt-tu-test_vit"