Added missing sensors

1 files changed, 2 insertions, 4 deletions

diff --git a/src/spacebot.cpp b/src/spacebot.cpp
index 23cce0c..15f2221 100644
--- a/src/spacebot.cpp
+++ b/src/spacebot.cpp
@@ -20,14 +20,12 @@ void Spacebot::writeNextMove()
 Move Spacebot::chooseMove()
 {
     auto sensorInputs = _gameState.toBitArray();
-    std::cout << sensorInputs.size() << std::endl;
     
     NeuralNetwork network(std::ifstream(_brainFilename),
 			  sensorInputs,
-			  static_cast<int>(Move::BUILD_SHIELD));
+			  7);
 
-    int moveInt = network.findMaxOutputIndex();
-    return static_cast<Move>(moveInt);
+    return static_cast<Move>(network.findMaxOutputIndex());
 }
 
 void Spacebot::writeMove(const Move& move)