From 05d28c4f5b72612dcbbab799017d9958f8bc1e3b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Czeka=C5=84ski?= Date: Sat, 23 May 2020 23:47:36 +0200 Subject: [PATCH] Remove commas from decision tree's enums Commas were causing enum values to be tuples instead of integers. --- .../DecisionTrees/projectSpecificClasses/DistFromObject.py | 4 ++-- .../projectSpecificClasses/PlayerStatsValue.py | 6 +++--- .../projectSpecificClasses/SurvivalClassification.py | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py b/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py index 2c15130..23f9af9 100644 --- a/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py +++ b/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py @@ -7,6 +7,6 @@ class DistFromObject(Enum): """ - LT_3 = 0, - GE_3_LT_8 = 1, + LT_3 = 0 + GE_3_LT_8 = 1 GE_8 = 2 diff --git a/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py b/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py index 70600f5..1fa339c 100644 --- a/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py +++ b/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py @@ -2,7 +2,7 @@ from enum import Enum class PlayerStatsValue(Enum): - ZERO_TO_QUARTER = 0, - QUARTER_TO_HALF = 1, - HALF_TO_THREE_QUARTERS = 2, + ZERO_TO_QUARTER = 0 + QUARTER_TO_HALF = 1 + HALF_TO_THREE_QUARTERS = 2 THREE_QUARTERS_TO_FULL = 3 diff --git a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py index 6d813e3..d41ab79 100644 --- a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py +++ b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py @@ -6,6 +6,6 @@ class SurvivalClassification(Enum): For example food classification means that player should go for food. """ - FOOD = 0, - WATER = 1, + FOOD = 0 + WATER = 1 REST = 2