Fix bug in DF

sailfish009 · Oct 7, 2017 · a2afcfb · a2afcfb
1 parent 2bc9c41
commit a2afcfb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/elf/replay_loader.h b/elf/replay_loader.h
@@ -13,17 +13,12 @@ class ReplayLoaderT {
     using RBuffer = SharedReplayBuffer<K, Record>;
     using GenFunc = typename RBuffer::GenFunc;
 
-private:
-    // Shared buffer for OfflineLoader.
-    static std::unique_ptr<RBuffer> _rbuffer;
-
-    RIterator _it;
-
-public:
     static void Init(GenFunc func) {
         _rbuffer.reset(new RBuffer(func));
     }
 
+    void Init() { reload(); }
+
     const RIterator &curr() const { return _it; }
 
     void Next() {
@@ -36,6 +31,11 @@ class ReplayLoaderT {
     }
 
 private:
+    // Shared buffer for OfflineLoader.
+    static std::unique_ptr<RBuffer> _rbuffer;
+
+    RIterator _it;
+
     void reload() {
         while (true) {
             K k = get_key();

diff --git a/go/offpolicy_loader.cc b/go/offpolicy_loader.cc
@@ -10,6 +10,7 @@ string OfflineLoader::_path;
 
 OfflineLoader::OfflineLoader(const GameOptions &options, int seed)
     : _options(options), _game_loaded(0), _rng(seed) {
+      ReplayLoader::Init();
 }
 
 void OfflineLoader::InitSharedBuffer(const std::string &list_filename) {
@@ -98,7 +99,7 @@ std::string OfflineLoader::get_key() {
 }
 
 bool OfflineLoader::need_reload(const Sgf::iterator &it) const {
-   return (it.done() || it.StepLeft() < _options.num_future_actions 
+   return (it.done() || it.StepLeft() < _options.num_future_actions
             || (_options.move_cutoff >= 0 && it.GetCurrIdx() >= _options.move_cutoff));
 }