Versioning

Author: Vincent Moens

.github/scripts/td_script.sh

@@ -1,5 +1,6 @@
 #!/bin/bash
 
-export TORCHRL_BUILD_VERSION=0.7.1
+export TORCHRL_BUILD_VERSION=0.7.2
 
-${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U
+#${CONDA_RUN} pip install git+https://github.com/pytorch/tensordict.git -U
+pip install tensordict==0.7.2

.github/scripts/version_script.bat

@@ -1,5 +1,5 @@
 @echo off
-set TORCHRL_BUILD_VERSION=0.7.1
+set TORCHRL_BUILD_VERSION=0.7.2
 echo TORCHRL_BUILD_VERSION is set to %TORCHRL_BUILD_VERSION%
 
 @echo on

test/test_exploration.py

@@ -123,7 +123,7 @@ def test_egreedy_masked(self, module, eps_init, spec_class):
             {"observation": torch.zeros(*batch_size, action_size)},
             batch_size=batch_size,
         )
-        with pytest.raises(KeyError, match="Action mask key action_mask not found in"):
+        with pytest.raises(RuntimeError, match="Failed while executing module"):
             explorative_policy(td)
 
         torch.manual_seed(0)
@@ -182,9 +182,7 @@ def test_no_spec_error(
             batch_size=batch_size,
         )
 
-        with pytest.raises(
-            RuntimeError, match="spec must be provided to the exploration wrapper."
-        ):
+        with pytest.raises(RuntimeError, match="Failed while executing module"):
             explorative_policy(td)
 
     @pytest.mark.parametrize("module", [True, False])
@@ -201,9 +199,7 @@ def test_wrong_action_shape(self, module):
                 policy,
             )
         td = TensorDict({"observation": torch.zeros(10, 4)}, batch_size=[10])
-        with pytest.raises(
-            ValueError, match="Action spec shape does not match the action shape"
-        ):
+        with pytest.raises(RuntimeError, match="Failed while executing module"):
             explorative_policy(td)
 
 
@@ -383,9 +379,8 @@ def test_nested(
         )
 
         action_spec = env.action_spec
-        d_act = action_spec.shape[-1]
+        action_spec.shape[-1]
 
-        net = nn.LazyLinear(d_act).to(device)
         policy = TensorDictModule(
             CountingEnvCountModule(action_spec=action_spec),
             in_keys=[("data", "states") if nested_obs_action else "observation"],

version.txt

@@ -1 +1 @@
-0.7.1
+0.7.2