From 81133b3e27eef2cb3caad824c2e0d18b947b17b9 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Sat, 21 Sep 2024 05:15:35 +0200
Subject: [PATCH] Update.

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 21666d1..5dceefc 100755
--- a/main.py
+++ b/main.py
@@ -399,7 +399,7 @@ def predict_the_four_grids(
     if with_noise:
         imt_set = add_noise_imt(imt_set)
 
-    result = ae_predict(model, imt_set, local_device=local_device, desc=None)
+    result = ae_predict(model, imt_set, local_device=local_device)
     result = (result * masks).reshape(-1, 4, result.size(1)).sum(dim=1)
 
     return result
-- 
2.39.5