From 1181ff0f63fb1b6c6e7429ccb9806a21b1103294 Mon Sep 17 00:00:00 2001 From: Yin Li Date: Thu, 20 May 2021 18:14:00 -0400 Subject: [PATCH] Fix bug when dis is Tensor --- map2map/utils/figures.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/map2map/utils/figures.py b/map2map/utils/figures.py index bcdb792..316bbf6 100644 --- 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 = [], []