diff --git a/python-api-src/library.cpp b/python-api-src/library.cpp index 020208d32fe76584a587c4b48f96cde4639819d9..165d95ca927dc5e4deff846940ca9b45c2e55daf 100644 --- a/python-api-src/library.cpp +++ b/python-api-src/library.cpp @@ -50,6 +50,12 @@ PYBIND11_MODULE(library, m) py::implicitly_convertible<sc2::UPGRADE_ID, sc2::UpgradeID>(); + py::class_<sc2::AbilityID>(m, "AbilityID") + .def(py::init<sc2::ABILITY_ID>()); + + py::implicitly_convertible<sc2::ABILITY_ID, sc2::AbilityID>(); + py::implicitly_convertible<sc2::AbilityID, sc2::ABILITY_ID>(); + py::class_<sc2::Agent>(m, "Agent") .def(py::init());