From 7fcfd694ba30d9adea15705538b8b303e390072f Mon Sep 17 00:00:00 2001
From: Weiqiang Zhu <75299929+zhuwq0@users.noreply.github.com>
Date: Thu, 25 Jan 2024 01:34:40 +0000
Subject: [PATCH] update

---
 gamma/seismic_ops.py | 4 ++++
 setup.py             | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/gamma/seismic_ops.py b/gamma/seismic_ops.py
index b04713a..2f55f95 100644
--- a/gamma/seismic_ops.py
+++ b/gamma/seismic_ops.py
@@ -155,6 +155,8 @@ def traveltime(event_loc, station_loc, phase_type, eikonal):
     nz = eikonal["nz"]
     h = eikonal["h"]
 
+    if isinstance(phase_type, list):
+        phase_type = np.array(phase_type)
     p_index = phase_type == "p"
     s_index = phase_type == "s"
     tt = np.zeros(len(phase_type), dtype=np.float32)
@@ -175,6 +177,8 @@ def grad_traveltime(event_loc, station_loc, phase_type, eikonal):
     nz = eikonal["nz"]
     h = eikonal["h"]
 
+    if isinstance(phase_type, list):
+        phase_type = np.array(phase_type)
     p_index = phase_type == "p"
     s_index = phase_type == "s"
     dt_dr = np.zeros(len(phase_type))
diff --git a/setup.py b/setup.py
index 15cc53a..107e483 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="GMMA",
-    version="1.2.5",
+    version="1.2.6",
     long_description="*GaMMA*: *Ga*ussian *M*ixture *M*odel *A*ssociation",
     long_description_content_type="text/markdown",
     packages=["gamma"],