From 052184223657744a49d49682ad41eca80849ed10 Mon Sep 17 00:00:00 2001
From: Eric Benz <79341534+ebnz@users.noreply.github.com>
Date: Thu, 27 Nov 2025 19:24:43 +0100
Subject: [PATCH] Add hparams parameter to moment computation

---
 easyeditor/models/pmet/pmet_main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/easyeditor/models/pmet/pmet_main.py b/easyeditor/models/pmet/pmet_main.py
index 5112e0d7..23b0bc65 100644
--- a/easyeditor/models/pmet/pmet_main.py
+++ b/easyeditor/models/pmet/pmet_main.py
@@ -336,6 +336,7 @@ def get_cov(
             sample_size=mom2_n_samples,
             precision=mom2_dtype,
             force_recompute=force_recompute,
+            hparams=hparams
         )
         COV_CACHE[key] = stat.mom2.moment().float().to("cpu")