Expose offers to Python.

PiperOrigin-RevId: 474786169 Change-Id: I39994376d6afc03b539d6c9af02f7a32c85fff45
google-deepmind · Sep 17, 2022 · 7f9a27c · 7f9a27c
1 parent e943f29
commit 7f9a27c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/open_spiel/games/bargaining.h b/open_spiel/games/bargaining.h
@@ -115,9 +115,11 @@ class BargainingState : public State {
       int player_id, std::function<double()> rng) const override;
 
   // Extra methods not part of the general API.
-  Instance instance() const { return instance_; }
+  Instance GetInstance() const { return instance_; }
   void SetInstance(Instance instance);
 
+  std::vector<Offer> Offers() const { return offers_; }
+
   Action AgreeAction() const;
 
  protected:

diff --git a/open_spiel/python/pybind11/games_bargaining.cc b/open_spiel/python/pybind11/games_bargaining.cc
@@ -40,7 +40,8 @@ void open_spiel::init_pyspiel_games_bargaining(py::module& m) {
       .def_readwrite("quantities", &Offer::quantities);
 
   py::classh<BargainingState, State>(m, "BargainingState")
-      .def("instance", &BargainingState::instance)
+      .def("instance", &BargainingState::GetInstance)
+      .def("offers", &BargainingState::Offers)
       .def("agree_action", &BargainingState::AgreeAction)
       // set_instance(instance)
       .def("set_instance", &BargainingState::SetInstance)