diff --git a/.conda/meta.yaml b/.conda/meta.yaml index 11f8e544e..238bf1e9c 100644 --- 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 index 5bc3f1bed..173c412df 100644 --- 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 index 155ad1389..92b38870d 100644 --- 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 index fcc5504cf..d73b645dc 100644 --- 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 index 9d875c8e0..b5c9bea47 100644 --- 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 index 22d09810b..53393e700 100644 --- 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'