lint

test/test_specs.py

@@ -1215,13 +1215,7 @@ def test_ndbounded_shape(self):
 
 
 class TestExpand:
- @pytest.mark.parametrize(
- "shape1",
- [
- None,
- (4,),
- (5, 4)
- ])
+ @pytest.mark.parametrize("shape1", [None, (4,), (5, 4)])
  @pytest.mark.parametrize("shape2", [(), (10,)])
  def test_binary(self, shape1, shape2):
  spec = BinaryDiscreteTensorSpec(

torchrl/_utils.py

@@ -577,6 +577,7 @@ def get_trace():
 
 class _ProcessNoWarn(mp.Process):
  """A private Process class that shuts down warnings on the subprocess and controls the number of threads in the subprocess."""
+
  @wraps(mp.Process.__init__)
  def __init__(self, *args, num_threads=None, **kwargs):
  import torchrl

torchrl/collectors/collectors.py

@@ -1143,7 +1143,7 @@ def __init__(
  )
  self.closed = True
  if num_threads is None:
- num_threads = len(create_env_fn) + 1 # 1 more thread for this proc
+ num_threads = len(create_env_fn) + 1  # 1 more thread for this proc
  self.num_sub_threads = num_sub_threads
  self.num_threads = num_threads
  self.create_env_fn = create_env_fn

torchrl/envs/batched_envs.py

@@ -166,7 +166,7 @@ def __init__(
  super().__init__(device=None)
  self.is_closed = True
  if num_threads is None:
- num_threads = num_workers + 1 # 1 more thread for this proc
+ num_threads = num_workers + 1  # 1 more thread for this proc
  self.num_sub_threads = num_sub_threads
  self.num_threads = num_threads
  self._cache_in_keys = None