Internal changes and wrapped game implementation.

PiperOrigin-RevId: 564496561 Change-Id: I941d9950aee3557746867b3d0e2b88e2e5dd031c
google-deepmind · Sep 12, 2023 · 894710b · 894710b
1 parent 2847cef
commit 894710b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/open_spiel/game_transforms/game_wrapper.h b/open_spiel/game_transforms/game_wrapper.h
@@ -16,6 +16,7 @@
 #define OPEN_SPIEL_GAME_TRANSFORMS_GAME_WRAPPER_H_
 
 #include "open_spiel/spiel.h"
+#include "open_spiel/spiel_globals.h"
 
 // Wraps a game, forwarding everything to the original implementation.
 // Transforms can inherit from this, overriding only what they need.
@@ -131,6 +132,15 @@ class WrappedGame : public Game {
     return game_->ObservationTensorShape();
   }
 
+  TensorLayout InformationStateTensorLayout() const override {
+    return game_->InformationStateTensorLayout();
+  }
+  TensorLayout ObservationTensorLayout() const override {
+    return game_->ObservationTensorLayout();
+  }
+  std::vector<int> PolicyTensorShape() const override {
+    return game_->PolicyTensorShape();
+  }
   int MaxGameLength() const override { return game_->MaxGameLength(); }
   int MaxChanceNodesInHistory() const override {
     return game_->MaxChanceNodesInHistory();