From 19e6d241d5316eaf52c63419c60ef805880abd62 Mon Sep 17 00:00:00 2001 From: k4yt3x Date: Mon, 4 May 2020 17:19:48 -0400 Subject: [PATCH] fixing driver argument TypeError --- src/video2x.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/video2x.py b/src/video2x.py index cbed695..b9abab4 100755 --- a/src/video2x.py +++ b/src/video2x.py @@ -167,10 +167,11 @@ config = read_config(video2x_args.config) driver_settings = config[video2x_args.driver] # overwrite driver_settings with driver_args -driver_args_dict = vars(driver_args) -for key in driver_args_dict: - if driver_args_dict[key] is not None: - driver_settings[key] = driver_args_dict[key] +if driver_args is not None: + driver_args_dict = vars(driver_args) + for key in driver_args_dict: + if driver_args_dict[key] is not None: + driver_settings[key] = driver_args_dict[key] # check if driver path exists if not pathlib.Path(driver_settings['path']).exists():