From 35e7aab636d1c4f8ca23c25e1ef64a45c5f3995f Mon Sep 17 00:00:00 2001
From: OlaRonning <ronning@pm.me>
Date: Tue, 27 Feb 2024 11:38:41 +0100
Subject: [PATCH] fixed to 3.8 syntax

---
 numpyro/contrib/ecs_proxies.py | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/numpyro/contrib/ecs_proxies.py b/numpyro/contrib/ecs_proxies.py
index 421b981a4..0ed852af4 100644
--- a/numpyro/contrib/ecs_proxies.py
+++ b/numpyro/contrib/ecs_proxies.py
@@ -162,15 +162,14 @@ def log_likelihood_sum(params_flat, subsample_indices=None):
                 for k, v in log_likelihood(params_flat, subsample_indices).items()
             }
 
-        match degree:
-            case 2:
-                TPState = TaylorTwoProxyState
-            case 1:
-                TPState = TaylorOneProxyState
-            case _:
-                raise ValueError(
-                    "Taylor proxy only defined for first and second degree."
-                )
+        if degree == 2:
+            TPState = TaylorTwoProxyState
+        elif 1:
+            TPState = TaylorOneProxyState
+        else:
+            raise ValueError(
+                "Taylor proxy only defined for first and second degree."
+            )
 
         # those stats are dict keyed by subsample names
         ref_sum_log_lik = log_likelihood_sum(ref_params_flat)