Some style corrections

icepack · Jun 29, 2023 · 9e9f0c6 · 9e9f0c6
1 parent b339a52
commit 9e9f0c6
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/demo/lbb-stability/lbb-stability.py b/demo/lbb-stability/lbb-stability.py
@@ -91,15 +91,17 @@ def run(*, dimension, num_cells, element, num_layers=None, vdegree=None):
 parser.add_argument("--dimension", type=int, choices=[2, 3])
 elements = ["taylor-hood", "crouzeix-raviart", "mini"]
 parser.add_argument("--element", choices=elements, default="taylor-hood")
-parser.add_argument("--log-dx-min", type=int, default=2)
-parser.add_argument("--log-dx-max", type=int, default=5)
+parser.add_argument("--num-cells-min", type=int, default=4)
+parser.add_argument("--num-cells-max", type=int, default=32)
+parser.add_argument("--num-cells-step", type=int, default=1)
 parser.add_argument("--num-layers", type=int, default=1)
 parser.add_argument("--vdegree", type=int, default=1)
 parser.add_argument("--output")
 args = parser.parse_args()
 
 stability_constants = []
-for num_cells in tqdm.trange(2**args.log_dx_min, 2**args.log_dx_max + 1):
+nmin, nmax, nstep = args.num_cells_min, args.num_cells_max, args.num_cells_step
+for num_cells in tqdm.trange(nmin, nmax + 1, nstep):
     h, λ = run(
         dimension=args.dimension,
         num_cells=num_cells,