Fix bug when dis is Tensor

2021-05-20 18:14:00 -04:00 · 2021-05-20 18:14:00 -04:00 · 1181ff0f63
commit 1181ff0f63
parent 6a3990f90c
1 changed files with 1 additions and 1 deletions
--- a/map2map/utils/figures.py
+++ b/map2map/utils/figures.py
@ -141,7 +141,7 @@ def plt_power(*fields, dis=None, label=None, **kwargs):
        label = [None] * len(fields)

    with torch.no_grad():
-        if dis:
+        if dis is not None:
            fields = lag2eul(dis, val=fields, **kwargs)

        ks, Ps = [], []