Assert

rdbende · Mar 18, 2023 · 1efbcff · 1efbcff
1 parent 91f2b6c
commit 1efbcff
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sv_ttk.py b/sv_ttk.py
@@ -36,11 +36,13 @@ def set_theme(theme):
     if theme not in {"dark", "light"}:
         raise RuntimeError("not a valid theme name: {}".format(theme))
 
+    assert root is not None
     root.tk.call("set_theme", theme)
 
 
 @init_theme
 def get_theme():
+    assert root is not None
     theme = root.tk.call("ttk::style", "theme", "use")
 
     return {"sun-valley-dark": "dark", "sun-valley-light": "light"}.get(theme, theme)