From b54e00fbe39d2fc70e67872029b3cbc2f3519e17 Mon Sep 17 00:00:00 2001 From: Dave O'Connor Date: Tue, 19 Jan 2021 14:04:07 +0000 Subject: [PATCH] Add none-checking to schedule_format_int_to_enum --- neohubapi/enums.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/neohubapi/enums.py b/neohubapi/enums.py index 6e88b49..0ea4f1c 100644 --- a/neohubapi/enums.py +++ b/neohubapi/enums.py @@ -23,6 +23,8 @@ class ScheduleFormat(enum.Enum): def schedule_format_int_to_enum(int_format): + if int_format is None: + return None if int_format == 0: return ScheduleFormat.ZERO elif int_format == 1: