fix

2024-02-04 20:03:31 +08:00
parent dca8c54210
commit 3ed870223b
1 changed files with 2 additions and 2 deletions
--- 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