From 83f4b69cf9f6fae866b3fb4b8557bb9738661fed Mon Sep 17 00:00:00 2001 From: rhoadesScholar Date: Thu, 9 Jan 2025 16:59:51 -0500 Subject: [PATCH] chore: Bump version to 0.1.2 in pyproject.toml and update input_size_step to accept a list in Architecture class --- pyproject.toml | 2 +- src/cellmap_models/pytorch/cosem/load_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index a9a20b2..f236e51 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,7 +15,7 @@ license = { text = "BSD 3-Clause License" } authors = [ { email = "rhoadesj@hhmi.org", name = "Jeff Rhoades" }, ] -version = "0.1.1" +version = "0.1.2" dependencies = [ 'torch', 'torchvision', diff --git a/src/cellmap_models/pytorch/cosem/load_model.py b/src/cellmap_models/pytorch/cosem/load_model.py index 05f52df..17c9523 100755 --- a/src/cellmap_models/pytorch/cosem/load_model.py +++ b/src/cellmap_models/pytorch/cosem/load_model.py @@ -252,7 +252,7 @@ def compute_minimal_shapes(self): self.min_input_shape = [int(s) for s in min_input_shape] self.min_output_shape = [int(s) for s in min_output_shape] - self.input_size_step = int(step) + self.input_size_step = [int(s) for s in step] def round_to_valid_input_shape(self, shape=None, mode="grow"): """