From 779369ecf681b359fb2a51404012dcf7b877bc93 Mon Sep 17 00:00:00 2001
From: wuaho <wuaho@legu.cc>
Date: Mon, 6 Sep 2021 12:01:34 +0800
Subject: [PATCH] 1

---
 common/compute.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/common/compute.py b/common/compute.py
index 38e7bd5..48f0479 100644
--- a/common/compute.py
+++ b/common/compute.py
@@ -1,9 +1,11 @@
 import numpy as np
+
+
 def division(a, b, n=2):
     res = 0
     try:
         res = round(a / b, n)
-        if np.isnan(res):
+        if np.isnan(res) or res == np.inf:
             res = 0
     except ZeroDivisionError:
         pass