Add missing enum imports (#157)

donnemartin · May 7, 2018 · a360483 · a360483
1 parent 4a7d2e4
commit a360483
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/solutions/object_oriented_design/online_chat/online_chat.py b/solutions/object_oriented_design/online_chat/online_chat.py
@@ -1,4 +1,5 @@
 from abc import ABCMeta
+from enum import Enum
 
 
 class UserService(object):

diff --git a/solutions/system_design/mint/mint_snippets.py b/solutions/system_design/mint/mint_snippets.py
@@ -1,12 +1,16 @@
 # -*- coding: utf-8 -*-
 
+from enum import Enum
+
+
 class DefaultCategories(Enum):
 
     HOUSING = 0
     FOOD = 1
     GAS = 2
     SHOPPING = 3
-    ...
+    # ...
+
 
 seller_category_map = {}
 seller_category_map['Exxon'] = DefaultCategories.GAS
@@ -44,4 +48,3 @@ def __init__(self, template_categories_to_budget_map):
 
     def override_category_budget(self, category, amount):
         self.categories_to_budget_map[category] = amount
-