fix: mypy

Author: iraedeus

mpest/models/beta.py

@@ -95,10 +95,10 @@ def generate(self, params: Params, size: int = 1, normalized: bool = True) -> Sa
         """
 
         if not normalized:
-            return beta.rvs(a=params[0], b=params[1], size=size)
+            return np.array(beta.rvs(a=params[0], b=params[1], size=size))
 
         a, b = self.params_convert_from_model(params)
-        return beta.rvs(a=a, b=b, size=size)
+        return np.array(beta.rvs(a=a, b=b, size=size))
 
     def pdf(self, x: float, params: Params) -> float:
         """Calculates the probability density function (PDF) value at :math:`x`.

mpest/models/pareto.py

@@ -97,10 +97,10 @@ def generate(self, params: Params, size: int = 1, normalized: bool = True) -> Sa
 
         if not normalized:
             x0, k = params
-            return pareto.rvs(b=k, scale=x0, size=size)
+            return np.array(pareto.rvs(b=k, scale=x0, size=size))
 
         x0, k = self.params_convert_from_model(params)
-        return pareto.rvs(b=k, scale=x0, size=size)
+        return np.array(pareto.rvs(b=k, scale=x0, size=size))
 
     def pdf(self, x: float, params: Params) -> float:
         r"""Calculate the probability density function (PDF) value at :math:`x`.