v0.3.2 (thu-ml#292)

Throw a warning in ListReplayBuffer. This version update is needed because of thu-ml#289, the previous v0.3.1 cannot work well under torch<=1.6.0 with cuda environment.
BFAnas · Feb 16, 2021 · 68ddc0c · 68ddc0c
1 parent 812560a
commit 68ddc0c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tianshou/__init__.py b/tianshou/__init__.py
@@ -1,7 +1,7 @@
 from tianshou import data, env, utils, policy, trainer, exploration
 
 
-__version__ = "0.3.1"
+__version__ = "0.3.2"
 
 __all__ = [
     "env",

diff --git a/tianshou/data/buffer.py b/tianshou/data/buffer.py
@@ -1,5 +1,6 @@
 import h5py
 import torch
+import warnings
 import numpy as np
 from numbers import Number
 from typing import Any, Dict, List, Tuple, Union, Optional
@@ -412,6 +413,7 @@ class ListReplayBuffer(ReplayBuffer):
 
     def __init__(self, **kwargs: Any) -> None:
         super().__init__(size=0, ignore_obs_next=False, **kwargs)
+        warnings.warn("ListReplayBuffer will be removed in version 0.4.0.")
 
     def sample(self, batch_size: int) -> Tuple[Batch, np.ndarray]:
         raise NotImplementedError("ListReplayBuffer cannot be sampled!")