Simply added expanded attribute to pydantic model

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):

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(