From b59b21978cb75378b0914f6099e45b51460301e9 Mon Sep 17 00:00:00 2001
From: zhangyiqin <312065559@qq.com>
Date: Thu, 4 Dec 2025 09:26:47 +0800
Subject: [PATCH] Fix static annotation

---
 src/probeinterface/generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/probeinterface/generator.py b/src/probeinterface/generator.py
index 95b3b1ce..e704ea6c 100644
--- a/src/probeinterface/generator.py
+++ b/src/probeinterface/generator.py
@@ -101,7 +101,7 @@ def generate_tetrode(r: float = 10.0) -> Probe:
 
 def generate_multi_columns_probe(
     num_columns: int = 3,
-    num_contact_per_column: int = 10,
+    num_contact_per_column: int | list[int] = 10,
     xpitch: float = 20,
     ypitch: float = 20,
     y_shift_per_column: Optional[np.array | list] = None,