diff --git a/deeplabcut/utils/make_labeled_video.py b/deeplabcut/utils/make_labeled_video.py index 690d658314..d318de6f1c 100644 --- a/deeplabcut/utils/make_labeled_video.py +++ b/deeplabcut/utils/make_labeled_video.py @@ -124,7 +124,7 @@ def CreateVideo( nindividuals = len(Dataframe.columns.get_level_values("individuals").unique()) map2bp = [bplist.index(bp) for bp in all_bpts] nbpts_per_ind = ( - Dataframe.groupby(level="individuals", axis=1).size().values[0] // 3 + Dataframe.groupby(level="individuals", axis=1).size().values // 3 ) map2id = [] for i, j in enumerate(nbpts_per_ind): diff --git a/requirements.txt b/requirements.txt index 26c872ea7c..dbcdd18998 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ matplotlib<=3.5.2 napari-deeplabcut>=0.0.6 networkx>=2.6 numpy>=1.18.5 -pandas>=1.0.1 +pandas>=1.0.1,!=1.5.0 pyyaml qdarkstyle==3.1 scikit-image>=0.17 diff --git a/setup.py b/setup.py index 31bfe51f8c..eea5312652 100644 --- a/setup.py +++ b/setup.py @@ -33,7 +33,7 @@ "matplotlib>=3.3", "networkx>=2.6", "numpy>=1.18.5", - "pandas>=1.0.1", + "pandas>=1.0.1,!=1.5.0", "scikit-image>=0.17", "scikit-learn>=1.0", "scipy>=1.4",