Minor cleanup.

[python_utils.git] / site_config.py

diff --git a/site_config.py b/site_config.py
index 4968523d54732802e2a12d061e0ec27a9c5b557c..1281661abb6ac52d9fd6384916ce617731ad34b3 100644 (file)
--- a/site_config.py
+++ b/site_config.py
@@ -12,15 +12,15 @@ from type.locations import Location
 logger = logging.getLogger(__name__)
 
 args = config.add_commandline_args(
-    f'({__file__})',
-    'Args related to __file__'
+    f'Global Site Config ({__file__})',
+    'Args related to global site-specific configuration'
 )
 args.add_argument(
     '--site_config_override_location',
     default='NONE',
     const='NONE',
     nargs='?',
-    choices=('HOUSE', 'CABIN', 'NONE'),
+    choices=['HOUSE', 'CABIN', 'NONE'],
     help='Where are we, HOUSE, CABIN?  Overrides standard detection code.',
 )
 
@@ -33,7 +33,7 @@ class SiteConfig(object):
     network_netmask: str
     network_router_ip: str
     presence_location: Location
-    is_anyone_present: Callable[None, bool]
+    is_anyone_present: Callable
     arper_minimum_device_count: int
 
 
@@ -53,7 +53,7 @@ def get_location():
     """
     Returns location as an enum instead of a string.
 
-    >>> from locations import Location
+    >>> from type.locations import Location
     >>> location = get_location()
     >>> location == Location.HOUSE or location == Location.CABIN
     True