utdquake 0.2__tar.gz → 0.2.2__tar.gz

{utdquake-0.2/utdquake.egg-info → utdquake-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: utdquake
-Version: 0.2
+Version: 0.2.2
 Summary: University of Texas at Dallas Earthquake Dataset
 Home-page: https://github.com/ecastillot/UTDQuake
 Author: ecastillot (Emmanuel Castillo)
@@ -48,6 +48,8 @@ Dynamic: summary
 # <span style="background:#E87500; color:white; padding:2px 6px; border-radius:6px;">UTD</span>Quake
 -->
 
+[![Hugging Face Dataset](https://img.shields.io/badge/HuggingFace-Dataset-yellow?style=for-the-badge&logo=huggingface&logoColor=black)](https://huggingface.co/datasets/ecastillot/UTDQuake)  [![Read the Docs](https://img.shields.io/badge/Docs-Read%20the%20Docs-blue?style=for-the-badge&logo=read-the-docs)](https://utdquake.readthedocs.io/en/latest/index.html) 
+
 **University of Texas at Dallas Earthquake Dataset**
 
 A global earthquake dataset constructed from high-quality source and receiver metadata, including associated seismic phase picks across diverse station geometries.

{utdquake-0.2 → utdquake-0.2.2}/README.md

@@ -8,6 +8,8 @@
 # <span style="background:#E87500; color:white; padding:2px 6px; border-radius:6px;">UTD</span>Quake
 -->
 
+[![Hugging Face Dataset](https://img.shields.io/badge/HuggingFace-Dataset-yellow?style=for-the-badge&logo=huggingface&logoColor=black)](https://huggingface.co/datasets/ecastillot/UTDQuake)  [![Read the Docs](https://img.shields.io/badge/Docs-Read%20the%20Docs-blue?style=for-the-badge&logo=read-the-docs)](https://utdquake.readthedocs.io/en/latest/index.html) 
+
 **University of Texas at Dallas Earthquake Dataset**
 
 A global earthquake dataset constructed from high-quality source and receiver metadata, including associated seismic phase picks across diverse station geometries.

{utdquake-0.2 → utdquake-0.2.2}/utdquake/__init__.py

@@ -20,7 +20,7 @@ Usage
 >>> net.events.head()
 
 """
-__version__ = "0.2"
+__version__ = "0.2.2"
 
 from .core.utdquake import Dataset, Network
 from .core.data import download_snapshot, load

{utdquake-0.2 → utdquake-0.2.2}/utdquake/bank/utils.py

@@ -1721,6 +1721,10 @@ def destination_point(lat1_deg, lon1_deg, azimuth_deg, distance_deg):
     spherical forward problem
     Compute destination lat/lon given a starting point, azimuth, and distance in degrees.
 
+    see here: 
+    - https://edwilliams.org/avform147.htm#LL
+    - Snyder, J. P. 1926-1997. (1987). Map projections--a working manual. Washington: U.S. G.P.O. . Page 31 (page 45-scanned version)
+
     Args:
         lat1_deg: Latitude of starting point (degrees)
         lon1_deg: Longitude of starting point (degrees)

{utdquake-0.2 → utdquake-0.2.2}/utdquake/core/utdquake.py

@@ -107,8 +107,8 @@ class Dataset:
         str
             Summary of networks, stations, and events.
         """
-        return get_network_summary(stations=self.get_stations(), 
-                                    events= self.get_events())
+        return get_network_summary(stations=self.stations, 
+                                    events= self.events)
 
     @property
     def networks(self):
@@ -160,6 +160,7 @@ class Dataset:
                     include_stations: bool = False,
                     include_picks: bool = False,
                     include_banks: bool = False,
+                    include_travel_time: bool = False,
                     unzip_banks: bool = True,
                     overwrite: bool = False,
                 ) -> Path:
@@ -183,6 +184,8 @@ class Dataset:
             Whether to download seismic picks. Default: False.
         include_banks : bool, optional
             Whether to download the bank (synthetic) data. Default: False.
+        include_travel_time : bool, optional
+            Whether to download the travel time models. Default: False.
         overwrite : bool, optional
             If True, existing files will be re-downloaded. Default: False.
         unzip_banks : bool, optional
@@ -211,6 +214,7 @@ class Dataset:
             include_stations=include_stations,
             include_picks=include_picks,
             include_banks=include_banks,
+            include_travel_time=include_travel_time,
             overwrite=overwrite,
             unzip_banks=unzip_banks
         )
@@ -499,7 +503,7 @@ class Network:
     Parameters
     ----------
     name : str
-        Network name (e.g. "tx", "uw", "GCI").
+        Network name (e.g. "tx", "uw", "UWF").
     das : bool, optional
         Whether to use the DAS or not. If so, tou may define DAS cache root environment variable (`UTDQUAKE_DAS_ROOT`).
     """
@@ -590,7 +594,12 @@ class Network:
         paths = resolve_network_paths(self.name,das=self.das, include_stations=True)
         return pd.read_parquet(paths["stations"])
 
-    
+    @property
+    def travel_time(self) -> TravelTimeModel:
+        """Return the TravelTimeModel for this network."""
+        paths = resolve_network_paths(self.name,das=self.das, include_travel_time=True)
+        return TravelTimeModel.load(paths["travel_time"])
+
     def compute_stats(
                     self,
                     use_cache=True,

{utdquake-0.2 → utdquake-0.2.2}/utdquake/qc/travel_time.py

@@ -775,14 +775,16 @@ class TravelTimeModel:
         ----------
         phase : str
             Phase name.
-        distance : numpy.ndarray
-            Distances.
+        distance : int, float or numpy.ndarray
+            Distances at which to predict.
 
         Returns
         -------
         pandas.DataFrame
             Predicted sigma and percentile curves.
         """
+        if isinstance(distance, (int, float)):
+            distance = np.array([distance])
         return self.get_model(phase).predict(distance)
 
     # ---------------------------------------------------------------------

{utdquake-0.2 → utdquake-0.2.2}/utdquake/writers/schema.py

@@ -70,7 +70,7 @@ PREF_STATIONS_TYPES = {
 PREF_STATIONS_ORDER = list(PREF_STATIONS_TYPES.keys())
 
 PREF_EVENTS_ORDER = ["network",
-                     "time",
+                    "time",
                     "latitude",
                     "longitude",
                     "depth",
@@ -89,6 +89,7 @@ PREF_PICKS_ORDER = [
                     "azimuth",
                     "evaluation_mode",
                     "event_id",
+                    "pick_agency",
                     "origin_time",
                     ]
 
@@ -172,7 +173,7 @@ for k,v in PREF_PICKS_QC_TYPES.items():
     PREF_PICKS_TYPES[k].extend(v)
 
 # Add extra columns to picks types
-PREF_PICKS_TYPES["string_cols"].extend(["network","event_id",])
+PREF_PICKS_TYPES["string_cols"].extend(["network","event_id","pick_agency"])
 PREF_PICKS_TYPES["float_cols"].extend(["travel_time","linear_hyp_distance"])
 
 # Preferred event types
@@ -356,7 +357,7 @@ def sanitize_dataframe(
         print(f"Step 6: Forced boolean columns: {bool_cols}")
     
     # ---- 7. Auto-handle remaining object columns ----
-    for c in df.select_dtypes(include=["object"]).columns:
+    for c in df.select_dtypes(include=["object","string"]).columns:
         df[c] = df[c].astype("string")
     
     # ---- 8. Numeric columns → float64 for Arrow safety ----

{utdquake-0.2 → utdquake-0.2.2/utdquake.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: utdquake
-Version: 0.2
+Version: 0.2.2
 Summary: University of Texas at Dallas Earthquake Dataset
 Home-page: https://github.com/ecastillot/UTDQuake
 Author: ecastillot (Emmanuel Castillo)
@@ -48,6 +48,8 @@ Dynamic: summary
 # <span style="background:#E87500; color:white; padding:2px 6px; border-radius:6px;">UTD</span>Quake
 -->
 
+[![Hugging Face Dataset](https://img.shields.io/badge/HuggingFace-Dataset-yellow?style=for-the-badge&logo=huggingface&logoColor=black)](https://huggingface.co/datasets/ecastillot/UTDQuake)  [![Read the Docs](https://img.shields.io/badge/Docs-Read%20the%20Docs-blue?style=for-the-badge&logo=read-the-docs)](https://utdquake.readthedocs.io/en/latest/index.html) 
+
 **University of Texas at Dallas Earthquake Dataset**
 
 A global earthquake dataset constructed from high-quality source and receiver metadata, including associated seismic phase picks across diverse station geometries.