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]>
talmolab · Nov 2, 2024 · c5e3ce8 · c5e3ce8
1 parent 63f2d8d
commit c5e3ce8
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/.conda/meta.yaml b/.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
 

diff --git a/.conda_mac/meta.yaml b/.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
diff --git a/environment.yml b/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

diff --git a/environment_mac.yml b/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]"
diff --git a/environment_no_cuda.yml b/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:

diff --git a/pypi_requirements.txt b/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'