From 92050fa35139d9850c7dc40f118d4d79d17f8da3 Mon Sep 17 00:00:00 2001
From: Alex Wernle <awernle@usgs.gov>
Date: Wed, 11 Dec 2024 16:33:55 -0700
Subject: [PATCH] Changed metadata_class to typehint of MetadataCategory since
this will not be used for flags only.
---
geomagio/algorithm/MetadataAlgorithm.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/geomagio/algorithm/MetadataAlgorithm.py b/geomagio/algorithm/MetadataAlgorithm.py
index 027e753d..43b96804 100644
--- a/geomagio/algorithm/MetadataAlgorithm.py
+++ b/geomagio/algorithm/MetadataAlgorithm.py
@@ -4,7 +4,6 @@ from datetime import timedelta
from enum import Enum
from typing import Optional
-from ..metadata.flag.Flag import Flag
from ..metadata import Metadata, MetadataFactory, MetadataCategory
from ..edge.MiniSeedFactory import MiniSeedFactory
from ..edge.EdgeFactory import EdgeFactory
@@ -55,7 +54,7 @@ class MetadataAlgorithm(BaseModel):
def create_metadata(
self,
- metadata_class: Flag,
+ metadata_class: MetadataCategory,
metadata_dict: dict,
category: MetadataCategory,
network: str,
@@ -103,7 +102,7 @@ class MetadataAlgorithm(BaseModel):
return self._get_metadata_factory().create_metadata(metadata=metadata_obj)
def split_stream_by_day(self, stream: Stream) -> list[Stream]:
- """Split stream into daily streams to prevent metadata from overlapping."""
+ """Split stream into daily streams to control size of spike arrays."""
daily_streams = []
# get min and max time
current_time = min(trace.stats.starttime for trace in stream)
--
GitLab