From 3ed870223b2e7e109e930dd1d6563e7ca9885b27 Mon Sep 17 00:00:00 2001
From: Sgr A* VMT <1611902585@qq.com>
Date: Sun, 4 Feb 2024 20:03:31 +0800
Subject: [PATCH] fix

---
 idm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/idm.py b/idm.py
index dd2540a..76222ee 100644
--- a/idm.py
+++ b/idm.py
@@ -1856,8 +1856,8 @@ class IDMMeshHelper:
             inputs = np.array(start)
             inputs += [dx,dy]
             while ((inputs >= 0).all() and (inputs <= [xi_max,yi_max]).all()):
-                if clusters[tuple(inputs)] is not None:
-                    return (abs(inputs-start[0]).sum(), median(clusters[tuple(inputs)]))
+                if clusters.get(tuple(inputs),None) is not None:
+                    return (abs(inputs-np.array(start)).sum(), median(clusters[tuple(inputs)]))
                 inputs += [dx,dy]
             return None