From 9e6e6c717c2d6dc95f468e1782a7a9dfeacc57e2 Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Mon, 22 Aug 2022 21:29:13 +0800 Subject: [PATCH] Update requirements.txt --- tests/rs_models/test_seg_models.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/rs_models/test_seg_models.py b/tests/rs_models/test_seg_models.py index e2fe6a6..88fb6e1 100644 --- 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)]]