Browse Source

Add missing enum imports (#157)

cclauss 7 years ago
parent
commit
a36048346e

+ 1 - 0
solutions/object_oriented_design/online_chat/online_chat.py

@@ -1,4 +1,5 @@
 from abc import ABCMeta
+from enum import Enum
 
 
 class UserService(object):

+ 5 - 2
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 @@ class Budget(object):
 
     def override_category_budget(self, category, amount):
         self.categories_to_budget_map[category] = amount
-