fixed samples shape error

SURGroup · Apr 19, 2022 · 7c41dce · 7c41dce
1 parent fabe62a
commit 7c41dce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/UQpy/stochastic_process/KarhunenLoeveExpansion.py b/src/UQpy/stochastic_process/KarhunenLoeveExpansion.py
@@ -184,7 +184,7 @@ def run(self, nsamples, random_variables=None):
         The :meth:`run` method has no returns, although it creates and/or appends the :py:attr:`samples` attribute of
         the :class:`KarhunenLoeveExpansion` class.
         """
-        samples = np.zeros((nsamples, self.correlation_function.shape[0], self.correlation_function.shape[1]))
+        samples = np.zeros((nsamples, self.correlation_function.shape[0], self.correlation_function.shape[2]))
         if random_variables is None:
             random_variables = np.random.normal(size=[self.thresholds[1], self.thresholds[0], nsamples])
         else: