From e071bf8b6e1d0cc006f93870b67648ebe335f64d Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Thu, 4 Apr 2024 11:53:43 +0200
Subject: [PATCH] Simply added expanded attribute to pydantic model
---
model_server/base/roiset.py | 1 +
tests/test_roiset.py | 2 ++
2 files changed, 3 insertions(+)
diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py
index 59ed23b3..5193e61c 100644
--- a/model_server/base/roiset.py
+++ b/model_server/base/roiset.py
@@ -31,6 +31,7 @@ class PatchParams(BaseModel):
rgb_overlay_channels: List[Union[int, None]] = [None, None, None]
rgb_overlay_weights: List[float] = [1.0, 1.0, 1.0]
pad_to: int = 256
+ expanded: bool = False
class AnnotatedZStackParams(BaseModel):
diff --git a/tests/test_roiset.py b/tests/test_roiset.py
index 5bec7d84..de58f25f 100644
--- a/tests/test_roiset.py
+++ b/tests/test_roiset.py
@@ -409,6 +409,8 @@ class TestRoiSetMultichannelProducts(BaseTestRoiSetMonoProducts, unittest.TestCa
'pad_to': 256,
},
})
+ self.assertTrue(hasattr(p.patches_2d, 'pad_to'))
+ self.assertTrue(hasattr(p.patches_2d, 'expanded'))
where = output_path / 'run_exports_expanded_2d_patch'
res = self.roiset.run_exports(
--
GitLab