Skip to content

Commit dc3261f

Browse files
committed
chore: fix warnings in python script
1 parent a82dbb0 commit dc3261f

1 file changed

Lines changed: 28 additions & 19 deletions

File tree

core/threshold/NIST_scripts/non-threshold-parser.py

Lines changed: 28 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -33,16 +33,16 @@
3333

3434
class ResultEntry:
3535
def __init__(self):
36-
self.keygen_latency = -1
37-
self.keygen_memory = -1
38-
self.erc20_latency = -1
39-
self.erc20_memory = -1
40-
self.encrypt_latency = -1
41-
self.encrypt_memory = -1
42-
self.decrypt_latency = -1
43-
self.decrypt_memory = -1
44-
self.mul_latency = -1
45-
self.mul_memory = -1
36+
self.keygen_latency: float = -1
37+
self.keygen_memory: float = -1
38+
self.erc20_latency: float = -1
39+
self.erc20_memory: float = -1
40+
self.encrypt_latency: float = -1
41+
self.encrypt_memory: float = -1
42+
self.decrypt_latency: float = -1
43+
self.decrypt_memory: float = -1
44+
self.mul_latency: float = -1
45+
self.mul_memory: float = -1
4646

4747
RESULT_MAP = {
4848
"NIST_PARAMS_P32_SNS_FGLWE":ResultEntry(),
@@ -68,7 +68,7 @@ def find_op_from_json(data):
6868

6969
def parse_latency_keygen(data):
7070
parameters = find_parameters_from_json(data)
71-
if parameters == None:
71+
if parameters is None:
7272
return
7373
mean_latency = data["mean"]["estimate"]
7474
mean_unit = data["mean"]["unit"]
@@ -80,7 +80,7 @@ def parse_latency_keygen(data):
8080

8181
def parse_latency_erc20(data):
8282
parameters = find_parameters_from_json(data)
83-
if parameters == None:
83+
if parameters is None:
8484
return
8585
mean_latency = data["mean"]["estimate"]
8686
mean_unit = data["mean"]["unit"]
@@ -92,7 +92,7 @@ def parse_latency_erc20(data):
9292

9393
def parse_latency_basic_ops(data):
9494
parameters = find_parameters_from_json(data)
95-
if parameters == None:
95+
if parameters is None:
9696
return
9797
op = find_op_from_json(data)
9898
mean_latency = data["mean"]["estimate"]
@@ -153,31 +153,40 @@ def fetch_mean_memory(line):
153153

154154
def parse_memory_keygen(line):
155155
params = find_params_from_line(line)
156-
if params == None:
156+
if params is None:
157157
return
158-
(mean_memory,unit) = fetch_mean_memory(line)
158+
result = fetch_mean_memory(line)
159+
if result is None:
160+
return
161+
(mean_memory,unit) = result
159162

160163
memory = float(mean_memory) * UNIT_CONV_TO_KB[unit]
161164

162165
RESULT_MAP[params].keygen_memory = memory
163166

164167
def parse_memory_erc20(line):
165168
params = find_params_from_line(line)
166-
if params == None:
169+
if params is None:
167170
return
168171

169-
(mean_memory,unit) = fetch_mean_memory(line)
172+
result = fetch_mean_memory(line)
173+
if result is None:
174+
return
175+
(mean_memory,unit) = result
170176

171177
memory = float(mean_memory) * UNIT_CONV_TO_KB[unit]
172178

173179
RESULT_MAP[params].erc20_memory = memory
174180

175181
def parse_memory_basic_ops(line):
176182
params = find_params_from_line(line)
177-
if params == None:
183+
if params is None:
178184
return
179185

180-
(mean_memory,unit) = fetch_mean_memory(line)
186+
result = fetch_mean_memory(line)
187+
if result is None:
188+
return
189+
(mean_memory,unit) = result
181190

182191
memory = float(mean_memory) * UNIT_CONV_TO_KB[unit]
183192

0 commit comments

Comments
 (0)