Bugfix

probabilistic-numerics · Feb 17, 2023 · 8dc9a63 · 8dc9a63
1 parent b7cf0e3
commit 8dc9a63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/probnum/utils/arrayutils.py b/src/probnum/utils/arrayutils.py
@@ -10,9 +10,9 @@
 
 
 def atleast_1d(  # pylint: disable=missing-raises-doc
-    *rvs: Union[ArrayLike, scipy.sparse.spmatrix, probnum.randvars.RandomVariable],
+    *rvs: Union[ArrayLike, scipy.sparse.spmatrix, "probnum.randvars.RandomVariable"],
 ) -> Sequence[
-    Union[np.ndarray, scipy.sparse.spmatrix, probnum.randvars.RandomVariable]
+    Union[np.ndarray, scipy.sparse.spmatrix, "probnum.randvars.RandomVariable"]
 ]:
     """Convert arrays or random variables to arrays or random variables with at least
     one dimension.