IPKM/mdevaluate
6
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changed from len(index.shape) to index.ndim

Browse Source
This commit is contained in:
Sebastian Kloth 2023-12-28 13:33:47 +01:00
parent 4a6c02627a
commit f2806ca3ca
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/mdevaluate/correlation.py
View File

@ -99,9 +99,9 @@ def shifted_correlation(
return get_correlation(frames, start_frame, index, shifted_idx) return get_correlation(frames, start_frame, index, shifted_idx)
else: else:
index = selector(frames[start_frame]) index = selector(frames[start_frame])
if len(index.shape) == 1: if index.ndim == 1:
return get_correlation(frames, start_frame, index, shifted_idx) return get_correlation(frames, start_frame, index, shifted_idx)
elif len(index.shape) == 2: elif index.ndim == 2:
correlations = [] correlations = []
for ind in index: for ind in index:
correlations.append( correlations.append(
@ -110,7 +110,7 @@ def shifted_correlation(
return correlations return correlations
else: else:
raise ValueError( raise ValueError(
f"Index list of selector has {len(index.shape)} dimensions, " f"Index list of selector has {index.ndim} dimensions, "
"but should have 1 or 2" "but should have 1 or 2"
) )