[Docs] Fix typo

facebookresearch · Apr 17, 2024 · 6296a1f · 6296a1f
1 parent 3a9a40c
commit 6296a1f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarl/models/cnn.py b/benchmarl/models/cnn.py
@@ -55,8 +55,8 @@ class Cnn(Model):
 
     The BenchMARL CNN accepts multiple inputs of 2 types:
 
-    - images: Tensors of shape (*batch, X,Y,C)
-    - arrays: Tensors of shape (*batch, F)
+    - images: Tensors of shape ``(*batch,X,Y,C)``
+    - arrays: Tensors of shape ``(*batch,F)``
 
     The CNN model will check that all image inputs have the same shape (excluding the last dimension)
     and cat them along that dimension before processing them with :class:`torchrl.modules.ConvNet`.