From 87de2e7a6c3e47dfd1e3af8211182bcf4a5331e7 Mon Sep 17 00:00:00 2001
From: Krystyna Milian <krystyna.milian@gmail.com>
Date: Mon, 26 Feb 2024 23:39:10 +0100
Subject: [PATCH] restored default Phasenet phases labels

---
 scripts/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/models.py b/scripts/models.py
index bcb99dd..644696f 100644
--- a/scripts/models.py
+++ b/scripts/models.py
@@ -131,7 +131,7 @@ class PhaseNetLit(SeisBenchModuleLit):
         self.sigma = sigma
         self.sample_boundaries = sample_boundaries
         self.loss = vector_cross_entropy
-        self.model = sbm.PhaseNet(phases="PN", **kwargs)
+        self.model = sbm.PhaseNet(**kwargs)
 
     def forward(self, x):
         return self.model(x)