test

pytorch · Jan 23, 2025 · 2fb9b37 · 2fb9b37
1 parent e7e7ad7
commit 2fb9b37
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 9 deletions.
diff --git a/.github/workflows/reusable_tutorials.yml b/.github/workflows/reusable_tutorials.yml
@@ -27,7 +27,7 @@ on:
 jobs:
   tutorials:
     name: Run tutorials
-    runs-on: ubuntu-latest
+    runs-on: 4-core-ubuntu
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python

diff --git a/tutorials/test3.ipynb b/tutorials/test3.ipynb
@@ -58,6 +58,10 @@
     "    keep_feasible=True,\n",
     ")\n",
     "x0 = np.random.random(128)\n",
+    "\n",
+    "sortby = SortKey.CUMULATIVE\n",
+    "\n",
+    "\n",
     "None"
    ]
   },
@@ -195,7 +199,6 @@
     "        options={\"maxfun\": 20},\n",
     "    )\n",
     "\n",
-    "sortby = SortKey.CUMULATIVE\n",
     "ps = pstats.Stats(pr).sort_stats(sortby)\n",
     "ps.print_stats(.1)\n",
     "None"
@@ -234,7 +237,9 @@
     "\n",
     "with cProfile.Profile() as pr:\n",
     "    for _ in range(20):\n",
-    "        f_and_grad(X)"
+    "        f_and_grad(X)\n",
+    "\n",
+    "ps = pstats.Stats(pr).sort_stats(sortby)"
    ]
   },
   {
@@ -334,7 +339,6 @@
     "        options={\"maxfun\": 20},\n",
     "    )\n",
     "\n",
-    "sortby = SortKey.CUMULATIVE\n",
     "ps = pstats.Stats(pr).sort_stats(sortby)\n",
     "ps.print_stats(.1)\n",
     "None"
@@ -373,7 +377,9 @@
     "\n",
     "with cProfile.Profile() as pr:\n",
     "    for _ in range(20):\n",
-    "        f_and_grad(X)"
+    "        f_and_grad(X)\n",
+    "\n",
+    "ps = pstats.Stats(pr).sort_stats(sortby)\n"
    ]
   },
   {
@@ -470,7 +476,6 @@
     "        options={\"maxfun\": 20},\n",
     "    )\n",
     "\n",
-    "sortby = SortKey.CUMULATIVE\n",
     "ps = pstats.Stats(pr).sort_stats(sortby)\n",
     "ps.print_stats(.1)\n",
     "None"
@@ -509,7 +514,9 @@
     "\n",
     "with cProfile.Profile() as pr:\n",
     "    for _ in range(20):\n",
-    "        f_and_grad(X)"
+    "        f_and_grad(X)\n",
+    "\n",
+    "ps = pstats.Stats(pr).sort_stats(sortby)\n"
    ]
   },
   {
@@ -582,7 +589,6 @@
     "        options={\"maxfun\": 20},\n",
     "    )\n",
     "\n",
-    "sortby = SortKey.CUMULATIVE\n",
     "ps = pstats.Stats(pr).sort_stats(sortby)\n",
     "ps.print_stats(.1)\n",
     "None"
@@ -621,7 +627,9 @@
     "\n",
     "with cProfile.Profile() as pr:\n",
     "    for _ in range(20):\n",
-    "        f_and_grad(X)"
+    "        f_and_grad(X)\n",
+    "\n",
+    "ps = pstats.Stats(pr).sort_stats(sortby)\n"
    ]
   }
  ],