diff --git a/src/storage/labeling.cpp b/src/storage/labeling.cpp index 25aa4ba..65ebe18 100644 --- a/src/storage/labeling.cpp +++ b/src/storage/labeling.cpp @@ -32,5 +32,6 @@ void define_labeling(py::module& m) { ; - py::class_(m, "ChoiceLabeling", "Labeling for choices", labeling); + py::class_(m, "ChoiceLabeling", "Labeling for choices", labeling). + def("get_labels_of_choice", &storm::models::sparse::ChoiceLabeling::getLabelsOfChoice, py::arg("choice"), "get labels of a choice"); }