fix doc comments

src/pystencils/sympyextensions/random.py

@@ -14,7 +14,7 @@ from ..types import UserTypeSpec, create_type, PsIeeeFloatType, PsNamedArrayType
 class RngState(ABC):
     """State of an RNG object.
 
-    Instances of `RngStateBase` must be immutable, hashable, and comparable
+    Instances of `RngState` must be immutable, hashable, and comparable
     to facilitate comparison, hashing, and pickle serialization of RNG invocation expressions.
     """
 
@@ -107,7 +107,8 @@ class RngBase(ABC):
         """Get a symbolic invocation of the RNG and a symbol representing its result.
 
         Args:
-            counter: An expression specifying the external counter part of the RNG state, optional
+            counter: An expression specifying the external counter part of the RNG state,
+                e.g. the time step counter
         """
 
         rng_func, state = self._get_rng_func()