reijii/IDM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Sgr A* VMT
2024-02-04 20:03:31 +08:00
parent dca8c54210
commit 3ed870223b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
idm.py
View File

@@ -1856,8 +1856,8 @@ class IDMMeshHelper:
inputs = np.array(start) inputs = np.array(start)
inputs += [dx,dy] inputs += [dx,dy]
while ((inputs >= 0).all() and (inputs <= [xi_max,yi_max]).all()): while ((inputs >= 0).all() and (inputs <= [xi_max,yi_max]).all()):
if clusters[tuple(inputs)] is not None: if clusters.get(tuple(inputs),None) is not None:
return (abs(inputs-start[0]).sum(), median(clusters[tuple(inputs)])) return (abs(inputs-np.array(start)).sum(), median(clusters[tuple(inputs)]))
inputs += [dx,dy] inputs += [dx,dy]
return None return None