From 9e9f0c6bedcc42032f4c4ecf78896dbb95ef3b0d Mon Sep 17 00:00:00 2001 From: Daniel Shapero Date: Thu, 29 Jun 2023 11:52:00 -0700 Subject: [PATCH] Some style corrections --- demo/lbb-stability/lbb-stability.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/demo/lbb-stability/lbb-stability.py b/demo/lbb-stability/lbb-stability.py index cf736e1..74a567f 100644 --- 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,