Add support for torch=2.1.0 (#276)

durandtibo · Oct 5, 2023 · fdfd4e5 · fdfd4e5
1 parent 5876821
commit fdfd4e5
Show file tree

Hide file tree

Showing 4 changed files with 72 additions and 35 deletions.
diff --git a/.github/workflows/test-deps.yaml b/.github/workflows/test-deps.yaml
@@ -147,7 +147,7 @@ jobs:
       matrix:
         os: [ ubuntu-latest ]
         python-version: [ '3.9', '3.10', '3.11' ]
-        polars-version: [ '0.19.6', '0.18.15' ]
+        polars-version: [ '0.19.7', '0.18.15' ]
 
     steps:
       - name: Checkout
@@ -189,11 +189,11 @@ jobs:
       matrix:
         os: [ ubuntu-latest ]
         python-version: [ '3.9', '3.10', '3.11' ]
-        torch-version: [ 2.0.1, 1.13.1, 1.12.1, 1.11.0, 1.10.0 ]
+        torch-version: [ 2.1.0, 2.0.1, 1.13.1, 1.12.1, 1.11.0, 1.10.2 ]
         exclude:
-          - torch-version: 1.10.0
+          - torch-version: 1.10.2
             python-version: 3.10
-          - torch-version: 1.10.0
+          - torch-version: 1.10.2
             python-version: 3.11
           - torch-version: 1.11.0
             python-version: 3.11

diff --git a/README.md b/README.md
@@ -142,6 +142,7 @@ The following is the corresponding `coola` versions and supported dependencies.
 
 | `coola`  | `jax`<sup>*</sup> | `numpy`<sup>*</sup> | `pandas`<sup>*</sup> | `polars`<sup>*</sup> | `torch`<sup>*</sup> | `xarray`<sup>*</sup> | `python`      |
 |----------|-------------------|---------------------|----------------------|----------------------|---------------------|----------------------|---------------|
+| `0.0.24` | `>=0.3,<0.5`      | `>=1.21,<1.27`      | `>=1.3,<2.2`         | `>=0.18.3,<0.20`     | `>=1.10,<2.2`       | `>=2023.3,<2023.9`   | `>=3.9,<3.12` |
 | `0.0.23` | `>=0.3,<0.5`      | `>=1.21,<1.27`      | `>=1.3,<2.2`         | `>=0.18.3,<0.20`     | `>=1.10,<2.1`       | `>=2023.3,<2023.9`   | `>=3.9,<3.12` |
 | `0.0.22` | `>=0.3,<0.5`      | `>=1.20,<1.26`      | `>=1.3,<2.1`         | `>=0.18.3,<0.19`     | `>=1.10,<2.1`       | `>=2023.3,<2023.9`   | `>=3.9,<3.12` |
 | `0.0.21` | `>=0.3,<0.5`      | `>=1.20,<1.26`      | `>=1.3,<2.1`         | `>=0.18.3,<0.19`     | `>=1.10,<2.1`       | `>=2023.3,<2023.8`   | `>=3.9,<3.12` |

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "coola"
-version = "0.0.24a0"
+version = "0.0.24a1"
 description = "A library to check if two complex/nested objects are equal or not"
 readme = "README.md"
 authors = ["Thibaut Durand <[email protected]>"]
@@ -38,7 +38,7 @@ numpy = { version = ">=1.21,<1.27", optional = true }
 pandas = { version = ">=1.3,<2.2", optional = true }
 # polars: 0.18.3 is the minimal version because of https://github.com/pola-rs/polars/issues/9358
 polars = { version = ">=0.18.3,<0.20", optional = true }
-torch = { version = ">=1.10,<2.1", optional = true }
+torch = { version = ">=1.10,<2.2", optional = true }
 xarray = { version = ">=2023.3,<2023.9", optional = true }
 
 [tool.poetry.extras]