Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug when dis is Tensor

Browse Source
This commit is contained in:
Yin Li 2021-05-20 18:14:00 -04:00
parent 6a3990f90c
commit 1181ff0f63
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/utils/figures.py
View File

@ -141,7 +141,7 @@ def plt_power(*fields, dis=None, label=None, **kwargs):
label = [None] * len(fields) label = [None] * len(fields)
with torch.no_grad(): with torch.no_grad():
if dis: if dis is not None:
fields = lag2eul(dis, val=fields, **kwargs) fields = lag2eul(dis, val=fields, **kwargs)
ks, Ps = [], [] ks, Ps = [], []