diff --git a/tests/sweep_framework/sweeps/reduction/argmax/argmax.py b/tests/sweep_framework/sweeps/reduction/argmax/argmax.py
index aa91f3bf26f2..913d4178865d 100644
--- a/tests/sweep_framework/sweeps/reduction/argmax/argmax.py
+++ b/tests/sweep_framework/sweeps/reduction/argmax/argmax.py
@@ -26,9 +26,9 @@
 # Developers can create their own generator functions and pass them to the parameters as inputs.
 parameters = {
     "nightly": {
-        "input_shape": gen_shapes([1, 1, 1, 1], [2, 6, 128, 128], [1, 1, 1, 1], 128)
-        + gen_shapes([1, 1, 1], [6, 128, 128], [1, 1, 1], 128)
-        + gen_shapes([1, 1], [128, 128], [1, 1], 128),
+        "input_shape": gen_shapes([1, 1, 1, 1], [2, 6, 128, 128], [1, 1, 1, 1], 32)
+        + gen_shapes([1, 1, 1], [6, 128, 128], [1, 1, 1], 32)
+        + gen_shapes([1, 1], [128, 128], [1, 1], 32),
         "dim": [0, 1, 2, 3, None],
         "input_a_dtype": [ttnn.bfloat16, ttnn.bfloat8_b],
         "input_layout": [ttnn.ROW_MAJOR_LAYOUT, ttnn.TILE_LAYOUT],
diff --git a/tests/sweep_framework/sweeps/reduction/topk/topk.py b/tests/sweep_framework/sweeps/reduction/topk/topk.py
index 6eff551e9281..e38a9134e5c5 100644
--- a/tests/sweep_framework/sweeps/reduction/topk/topk.py
+++ b/tests/sweep_framework/sweeps/reduction/topk/topk.py
@@ -124,20 +124,3 @@ def run(
     )
 
     return [(passing, output_str), e2e_perf]
-
-
-from tests.sweep_framework.framework.permutations import *
-
-for suite in parameters.keys():
-    device_id = 0
-    device = ttnn.open_device(device_id=device_id)
-    suite_vectors = list(permutations(parameters[suite]))
-    print(len(suite_vectors))
-    for vector in suite_vectors:
-        if invalidate_vector(vector)[0]:
-            continue
-        passed, _ = run(**vector, device=device)
-        if passed[0] != True:
-            print(passed)
-
-    ttnn.close_device(device)