From ea4f2b1d8bacdaaefffc328870524bc12fb84138 Mon Sep 17 00:00:00 2001
From: mzouink <zouinkhi.marwan@gmail.com>
Date: Wed, 13 Nov 2024 14:09:30 -0500
Subject: [PATCH] fix dims error

---
 dacapo/experiments/architectures/architecture.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dacapo/experiments/architectures/architecture.py b/dacapo/experiments/architectures/architecture.py
index 0f188560e..7ec81bd4f 100644
--- a/dacapo/experiments/architectures/architecture.py
+++ b/dacapo/experiments/architectures/architecture.py
@@ -115,7 +115,7 @@ def dims(self) -> int:
         Note:
             The method is optional and can be overridden in the derived class.
         """
-        return self.input_shape.dims
+        return Coordinate(self.input_shape).dims
 
     def scale(self, input_voxel_size: Coordinate) -> Coordinate:
         """