Update settings.py

Serializing Enum member to JSON
datamllab · Apr 4, 2023 · 0a6a45d · 0a6a45d
1 parent d9a0007
commit 0a6a45d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rlcard/games/gin_rummy/utils/settings.py b/rlcard/games/gin_rummy/utils/settings.py
@@ -9,13 +9,13 @@
 from enum import Enum
 
 
-class DealerForRound(Enum):
+class DealerForRound(int, Enum):
  North = 0
  South = 1
  Random = 2
 
 
-class Setting(Enum):
+class Setting(str, Enum):
  dealer_for_round = "dealer_for_round"
  stockpile_dead_card_count = "stockpile_dead_card_count"
  going_out_deadwood_count = "going_out_deadwood_count"