From 99e8e58f82cff9b82d7455c576f5a237ce9a9a72 Mon Sep 17 00:00:00 2001 From: Benjamin Kiessling Date: Fri, 18 Oct 2024 16:35:26 +0200 Subject: [PATCH] s/add/union --- tests/test_newpolygons.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_newpolygons.py b/tests/test_newpolygons.py index 7c24ffb7b..119d99e7a 100644 --- a/tests/test_newpolygons.py +++ b/tests/test_newpolygons.py @@ -249,7 +249,7 @@ def test_ketoscli_train_old_model(self): fp = str(Path(tempdir) / "test.xml") self._test_ketoscli( - args=['train', '-f', 'xml', '-N', '1', '-q', 'fixed', '-i', self.old_model_path, '--resize', 'add', '-o', mfp, '--workers', '0', self.segmented_img], + args=['train', '-f', 'xml', '-N', '1', '-q', 'fixed', '-i', self.old_model_path, '--resize', 'union', '-o', mfp, '--workers', '0', self.segmented_img], expect_legacy=False, check_exit_code=[0, 1], # Model may not improve during training ) @@ -268,7 +268,7 @@ def test_ketoscli_train_old_model_force_legacy(self): fp = str(Path(tempdir) / "test.xml") self._test_ketoscli( - args=['train', '--legacy-polygons', '-f', 'xml', '-N', '1', '-q', 'fixed', '-i', self.old_model_path, '--resize', 'add', '-o', mfp, '--workers', '0', self.segmented_img], + args=['train', '--legacy-polygons', '-f', 'xml', '-N', '1', '-q', 'fixed', '-i', self.old_model_path, '--resize', 'union', '-o', mfp, '--workers', '0', self.segmented_img], expect_legacy=True, check_exit_code=[0, 1], # Model may not improve during training ) @@ -326,7 +326,7 @@ def test_ketoscli_pretrain_old_model(self): fp = str(Path(tempdir) / "test.xml") self._test_ketoscli( - args=['pretrain', '-f', 'xml', '-N', '1', '-q', 'fixed', '-i', self.old_model_path, '--resize', 'add', '-o', mfp, '--workers', '0', self.segmented_img], + args=['pretrain', '-f', 'xml', '-N', '1', '-q', 'fixed', '-i', self.old_model_path, '--resize', 'union', '-o', mfp, '--workers', '0', self.segmented_img], expect_legacy=False, check_exit_code=[0, 1], # Model may not improve during training ) @@ -349,7 +349,7 @@ def _assertWarnsWhenTrainingArrow(self, if force_legacy: args = ['--legacy-polygons'] + args if from_model: - args = ['-i', from_model, '--resize', 'add'] + args + args = ['-i', from_model, '--resize', 'union'] + args print("ketos", 'train', *args) run = self.runner.invoke(ketos_cli, ['train'] + args)