Lower bound albumentation version to handle Nan keypoints (#1995)

* Lower bound albumentation version to handle Nan keypoints

* Reset tabwidth to 2 in environment ymls

---------

Co-authored-by: Elizabeth <[email protected]>

Author: roomrys

.conda/meta.yaml

@@ -56,7 +56,7 @@ requirements:
     - sleap-deps::tensorflow ==2.9.2
     - conda-forge::tensorflow-hub
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0
     - conda-forge::importlib-metadata
   run:
@@ -91,7 +91,7 @@ requirements:
     - sleap-deps::tensorflow ==2.9.2
     - conda-forge::tensorflow-hub
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0
     - conda-forge::importlib-metadata
 

.conda_mac/meta.yaml

@@ -56,7 +56,7 @@ requirements:
     - conda-forge::seaborn
     - sleap-deps::tensorflow ==2.12.0
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0
 
   run:
@@ -87,5 +87,5 @@ requirements:
     - conda-forge::seaborn
     - sleap-deps::tensorflow ==2.12.0
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0

environment.yml

@@ -40,7 +40,7 @@ dependencies:
     - sleap-deps::tensorflow ==2.9.2
     - conda-forge::tensorflow-hub # Pinned in meta.yml, but no problems here... yet
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0
 
     # Packages required by tensorflow to find/use GPUs

environment_mac.yml

@@ -38,7 +38,7 @@ dependencies:
     - conda-forge::seaborn
     - sleap-deps::tensorflow ==2.12.0
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0
     - pip:
           - "--editable=.[conda_dev]"

environment_no_cuda.yml

@@ -40,7 +40,7 @@ dependencies:
     - sleap-deps::tensorflow ==2.9.2
     - conda-forge::tensorflow-hub # Pinned in meta.yml, but no problems here... yet
     - conda-forge::qudida
-    - conda-forge::albumentations
+    - conda-forge::albumentations >=1.4.15 # Handle Nan keypoints
     - conda-forge::ndx-pose <0.2.0
 
     - pip:

pypi_requirements.txt

@@ -30,7 +30,7 @@ scikit-learn>=1.0.0
 seaborn
 tensorflow==2.9.2; platform_machine != 'arm64'
 tensorflow-hub
-albumentations
+albumentations>=1.4.15 # Handle Nan keypoints
 ndx-pose<0.2.0
 # Silicon Mac specific packages
 tensorflow-macos >=2.10.0,<2.13.0; sys_platform == 'darwin' and platform_machine == 'arm64'