typing updates

This commit is contained in:
Jan Petykiewicz 2022-10-04 12:43:26 -07:00
commit d42a625e5f
2 changed files with 73 additions and 63 deletions

View file

@ -156,7 +156,7 @@ def test1(solver=generic_solver):
# grid.draw_cuboid(pmcg, center=[700, 0, 0], dimensions=[80, 1e8, 1e8], eps=1)
# grid.visualize_isosurface(pmcg)
def pcolor(v):
def pcolor(v) -> None:
vmax = numpy.max(numpy.abs(v))
pyplot.pcolor(v, cmap='seismic', vmin=-vmax, vmax=vmax)
pyplot.axis('equal')