Update requirements.txt

3 years ago · 9e6e6c717c
parent e4c0b553b7
commit 9e6e6c717c
1 changed files with 4 additions and 0 deletions
--- a/tests/rs_models/test_seg_models.py
+++ b/tests/rs_models/test_seg_models.py
@ -52,3 +52,7 @@ class TestFarSegModel(TestSegModel):
        self.specs = [
            dict(), dict(num_classes=20), dict(encoder_pretrained=False)
        ]
+
+    def set_targets(self):
+        self.targets = [[self.get_zeros_array(16)], [self.get_zeros_array(20)],
+                        [self.get_zeros_array(16)]]