[VCF] Some improvements in pot and extr_corr_utils

viccollado · viccollado · commit 6137212b27c9 · 2025-11-20T12:26:21.000+01:00
diff --git a/bluemath_tk/distributions/pot.py b/bluemath_tk/distributions/pot.py
@@ -318,7 +318,7 @@ def fit(
         _, _, _, self.pks, self.pks_idx, _ = pot(
             self.data, self.threshold, self.n0, self.min_peak_distance
         )
-        return self.threshold, self.pks, self.pks_idx
+        return self.threshold.item(), self.pks, self.pks_idx
 
     def studentized_residuals(
         self,
diff --git a/bluemath_tk/distributions/utils/extr_corr_utils.py b/bluemath_tk/distributions/utils/extr_corr_utils.py
@@ -30,7 +30,7 @@ def gpdpoiss_ci_rp_bootstrap(
     boot_params = np.zeros((B, 3))
     for i in range(B):
         fit_result = GPD.fit(boot_samples[i], threshold=threshold)
-        boot_params[i, :] = fit_result.params
+        boot_params[i, :] = np.asarray(fit_result.params)
 
     return_periods = np.array(
         [

Original file line number	Diff line number	Diff line change
`@@ -318,7 +318,7 @@ def fit(`
`318`	`318`	`_, _, _, self.pks, self.pks_idx, _ = pot(`
`319`	`319`	`self.data, self.threshold, self.n0, self.min_peak_distance`
`320`	`320`	`)`
`321`		`- return self.threshold, self.pks, self.pks_idx`
	`321`	`+ return self.threshold.item(), self.pks, self.pks_idx`
`322`	`322`
`323`	`323`	`def studentized_residuals(`
`324`	`324`	`self,`
Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ def gpdpoiss_ci_rp_bootstrap(`
`30`	`30`	`boot_params = np.zeros((B, 3))`
`31`	`31`	`for i in range(B):`
`32`	`32`	`fit_result = GPD.fit(boot_samples[i], threshold=threshold)`
`33`		`- boot_params[i, :] = fit_result.params`
	`33`	`+ boot_params[i, :] = np.asarray(fit_result.params)`
`34`	`34`
`35`	`35`	`return_periods = np.array(`
`36`	`36`	`[`