Fix docstrings

Author: sgreenbury

src/auto_cast/decoders/base.py

@@ -22,7 +22,7 @@ def decode(self, z: TensorBMStarL) -> TensorBTSPlusC:
 
         Parameters
         ----------
-        z: TensorBMultiL
+        z: TensorBMStarL
             Latent tensor to be decoded.
 
         Returns

src/auto_cast/encoders/base.py

@@ -30,7 +30,7 @@ def encode(self, batch: Batch) -> TensorBMStarL:
 
         Returns
         -------
-        TensorBMultiL
+        TensorBMStarL
             Encoded tensor in the latent space with shape (B, *, C_latent).
         """