chore: apply flake8-comprehensions Ruff rules to improve collection comprehensions (#5652)

Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang
2024-06-27 11:21:31 +08:00
committed by GitHub
parent 2e718b85e9
commit dcb72e0067
58 changed files with 123 additions and 136 deletions

View File

@@ -16,12 +16,12 @@ class BarChartTool(BuiltinTool):
data = data.split(';')
# if all data is int, convert to int
if all([i.isdigit() for i in data]):
if all(i.isdigit() for i in data):
data = [int(i) for i in data]
else:
data = [float(i) for i in data]
axis = tool_parameters.get('x_axis', None) or None
axis = tool_parameters.get('x_axis') or None
if axis:
axis = axis.split(';')
if len(axis) != len(data):

View File

@@ -17,14 +17,14 @@ class LinearChartTool(BuiltinTool):
return self.create_text_message('Please input data')
data = data.split(';')
axis = tool_parameters.get('x_axis', None) or None
axis = tool_parameters.get('x_axis') or None
if axis:
axis = axis.split(';')
if len(axis) != len(data):
axis = None
# if all data is int, convert to int
if all([i.isdigit() for i in data]):
if all(i.isdigit() for i in data):
data = [int(i) for i in data]
else:
data = [float(i) for i in data]

View File

@@ -16,10 +16,10 @@ class PieChartTool(BuiltinTool):
if not data:
return self.create_text_message('Please input data')
data = data.split(';')
categories = tool_parameters.get('categories', None) or None
categories = tool_parameters.get('categories') or None
# if all data is int, convert to int
if all([i.isdigit() for i in data]):
if all(i.isdigit() for i in data):
data = [int(i) for i in data]
else:
data = [float(i) for i in data]