Skip to content
Snippets Groups Projects
Commit 92050fa3 authored by Wernle, Alexandra Nicole's avatar Wernle, Alexandra Nicole
Browse files

Changed metadata_class to typehint of MetadataCategory since this will not be used for flags only.

parent b94c90b4
No related branches found
No related tags found
1 merge request!344New MetadataAlgorithm, SpikesAlgorithm, and flag-spikes application
...@@ -4,7 +4,6 @@ from datetime import timedelta ...@@ -4,7 +4,6 @@ from datetime import timedelta
from enum import Enum from enum import Enum
from typing import Optional from typing import Optional
from ..metadata.flag.Flag import Flag
from ..metadata import Metadata, MetadataFactory, MetadataCategory from ..metadata import Metadata, MetadataFactory, MetadataCategory
from ..edge.MiniSeedFactory import MiniSeedFactory from ..edge.MiniSeedFactory import MiniSeedFactory
from ..edge.EdgeFactory import EdgeFactory from ..edge.EdgeFactory import EdgeFactory
...@@ -55,7 +54,7 @@ class MetadataAlgorithm(BaseModel): ...@@ -55,7 +54,7 @@ class MetadataAlgorithm(BaseModel):
def create_metadata( def create_metadata(
self, self,
metadata_class: Flag, metadata_class: MetadataCategory,
metadata_dict: dict, metadata_dict: dict,
category: MetadataCategory, category: MetadataCategory,
network: str, network: str,
...@@ -103,7 +102,7 @@ class MetadataAlgorithm(BaseModel): ...@@ -103,7 +102,7 @@ class MetadataAlgorithm(BaseModel):
return self._get_metadata_factory().create_metadata(metadata=metadata_obj) return self._get_metadata_factory().create_metadata(metadata=metadata_obj)
def split_stream_by_day(self, stream: Stream) -> list[Stream]: 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 = [] daily_streams = []
# get min and max time # get min and max time
current_time = min(trace.stats.starttime for trace in stream) current_time = min(trace.stats.starttime for trace in stream)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment