Update torch dependency in tests

pyt-team · Jul 8, 2024 · 1e250ac · 1e250ac
1 parent 04d9437
commit 1e250ac
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 32 deletions.
diff --git a/.github/workflows/test_codebase.yml b/.github/workflows/test_codebase.yml
@@ -28,9 +28,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.3"]
 
     steps:
       - uses: actions/checkout@v3

diff --git a/.github/workflows/test_tutorials.yml b/.github/workflows/test_tutorials.yml
@@ -32,9 +32,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.3"]
         test-group: [1, 2, 3, 4]
 
     steps:
@@ -68,9 +66,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.3"]
         test-group: [1, 2, 3]
 
     steps:
@@ -104,9 +100,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.3"]
 
     steps:
       - uses: actions/checkout@v3
@@ -139,9 +133,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ["3.10", "3.11"]
-        torch-version: [2.0.1]
-        include:
-          - torch-version: 2.0.1
+        torch-version: ["2.3"]
 
     steps:
       - uses: actions/checkout@v3

diff --git a/tutorials/combinatorial/hmc_train.ipynb b/tutorials/combinatorial/hmc_train.ipynb
@@ -123,7 +123,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:49:28.758850145Z",
@@ -149,7 +149,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -178,7 +178,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:49:28.810114967Z",
@@ -237,16 +237,16 @@
     "        b2 = []\n",
     "\n",
     "        for cc in self.complexes:\n",
-    "            a0.append(torch.from_numpy(cc.adjacency_matrix(0, 1).todense()).to_sparse())\n",
-    "            a1.append(torch.from_numpy(cc.adjacency_matrix(1, 2).todense()).to_sparse())\n",
+    "            a0.append(torch.from_numpy(cc.adjacency_matrix(0, 1).todense()))\n",
+    "            a1.append(torch.from_numpy(cc.adjacency_matrix(1, 2).todense()))\n",
     "\n",
     "            B = cc.incidence_matrix(rank=1, to_rank=2)\n",
     "            A = B.T @ B\n",
     "            A.setdiag(0)\n",
-    "            coa2.append(torch.from_numpy(A.todense()).to_sparse())\n",
+    "            coa2.append(torch.from_numpy(A.todense()))\n",
     "\n",
-    "            b1.append(torch.from_numpy(cc.incidence_matrix(0, 1).todense()).to_sparse())\n",
-    "            b2.append(torch.from_numpy(cc.incidence_matrix(1, 2).todense()).to_sparse())\n",
+    "            b1.append(torch.from_numpy(cc.incidence_matrix(0, 1).todense()))\n",
+    "            b2.append(torch.from_numpy(cc.incidence_matrix(1, 2).todense()))\n",
     "\n",
     "        return a0, a1, coa2, b1, b2\n",
     "\n",
@@ -315,7 +315,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:49:29.818662234Z",
@@ -350,7 +350,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:50:47.902394227Z",
@@ -372,7 +372,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:55:26.795250682Z",
@@ -422,7 +422,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:55:26.814069014Z",
@@ -614,7 +614,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -681,7 +681,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T06:55:28.220563261Z",
@@ -712,7 +712,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -770,7 +770,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T07:01:54.496249166Z",
@@ -820,7 +820,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "metadata": {
     "ExecuteTime": {
      "end_time": "2023-08-24T07:13:58.198454432Z",