Add Non Stationary GEV class

bluemath_tk/distributions/__init__.py

@@ -9,6 +9,7 @@
 # Import essential functions/classes to be available at the package level.
 from .gev import GEV
 from .gpd import GPD
+from .nonstat_gev import NonStatGEV
 
 # Optionally, define the module's `__all__` variable to control what gets imported when using `from module import *`.
-__all__ = ["GEV", "GPD"]
+__all__ = ["GEV", "GPD", "NonStatGEV"]

bluemath_tk/distributions/_base_distributions.py

@@ -667,8 +667,12 @@ def obj(free_values):
             free_idx += 1
 
     # Create modified result with full parameters
+    # TODO: Modify the OptimizeResult to include all the attributes from the original result
+    # but with the full parameter vector
+    # TODO: Add hess info or hess_inv info to compute confidence intervals in FitResult
     modified_result = OptimizeResult(
-        x=full_params, success=result.success, fun=result.fun, message=result.message
+        x=full_params, success=result.success, fun=result.fun, message=result.message,
+        hess_inv=result.hess_inv if hasattr(result, 'hess_inv') else None
     )
 
     # Return the fitting result as a FitResult object