mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-07-06 14:16:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Speed up YAML parsing with CSafeLoader (#24958)

Browse Source 
Use CSafeLoader
old-commit-hash: 10fb2b9456b9d0d065b411ac64fd97831ee2fa8c
This commit is contained in:
Shane Smiskol
2022-06-24 15:16:46 -07:00
committed by GitHub
parent 3e2828c9bb
commit 275b7eac20
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/car/interfaces.py
+5 -5
View File
@@ -26,16 +26,16 @@ TORQUE_OVERRIDE_PATH = os.path.join(BASEDIR, 'selfdrive/car/torque_data/override
TORQUE_SUBSTITUTE_PATH = os.path.join(BASEDIR, 'selfdrive/car/torque_data/substitute.yaml')
def get_torque_params(candidate, default=float('NaN')):
def get_torque_params(candidate):
with open(TORQUE_SUBSTITUTE_PATH) as f:
sub = yaml.load(f, Loader=yaml.FullLoader)
sub = yaml.load(f, Loader=yaml.CSafeLoader)
if candidate in sub:
candidate = sub[candidate]
with open(TORQUE_PARAMS_PATH) as f:
params = yaml.load(f, Loader=yaml.FullLoader)
params = yaml.load(f, Loader=yaml.CSafeLoader)
with open(TORQUE_OVERRIDE_PATH) as f:
override = yaml.load(f, Loader=yaml.FullLoader)
override = yaml.load(f, Loader=yaml.CSafeLoader)
# Ensure no overlap
if sum([candidate in x for x in [sub, params, override]]) > 1:
@@ -47,7 +47,7 @@ def get_torque_params(candidate, default=float('NaN')):
out = params[candidate]
else:
raise NotImplementedError(f"Did not find torque params for {candidate}")
return {key:out[i] for i, key in enumerate(params['legend'])}
return {key: out[i] for i, key in enumerate(params['legend'])}
# generic car and radar interfaces