From eb71b9083e733c18aedc02ed13568aa32b1fe770 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hannes=20J=C3=A4mtner?= <hanja189@student.liu.se>
Date: Wed, 15 Jul 2020 11:43:52 +0200
Subject: [PATCH] Updated the library.cpp
---
python-api-src/library.cpp | 12 ++++++++++--
1 file changed, 10 insertions(+), 2 deletions(-)
diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp
index 9b231c4a3..e183d8d49 100644
--- a/python-api-src/library.cpp
+++ b/python-api-src/library.cpp
@@ -127,8 +127,16 @@ PYBIND11_MODULE(library, m)
.value("CheatMoney", sc2::Difficulty::CheatMoney)
.value("CheatInsane", sc2::Difficulty::CheatInsane);
- m.def("create_participants", &sc2::CreateParticipant, "Create participant from bot", "race"_a, "bot"_a, "player_name"_a);
- m.def("create_computer", &sc2::CreateComputer, "Create participant from built-in Starcraft computer", "race"_a, "difficulty"_a, "ai_build"_a, "player_name"_a);
+ py::enum_<sc2::AIBuild>(m, "AIBuild")
+ .value("RandomBuild", sc2::AIBuild::RandomBuild)
+ .value("Rush", sc2::AIBuild::Rush)
+ .value("Timing", sc2::AIBuild::Timing)
+ .value("Power", sc2::AIBuild::Power)
+ .value("Macro", sc2::AIBuild::Air);
+
+
+ m.def("create_participants", &sc2::CreateParticipant, "Create participant from bot", "race"_a, "bot"_a, "player_name"_a = "");
+ m.def("create_computer", &sc2::CreateComputer, "Create participant from built-in Starcraft computer", "race"_a, "difficulty"_a, "ai_build"_a = sc2::AIBuild::RandomBuild, "player_name"_a = "");
/*
py::class_<BuildDescription>(m, "BuildDescription")
--
GitLab