diff --git a/geomagio/processing/factory.py b/geomagio/processing/factory.py index 6e741d469110eedcf02993273ffb12ca02b01cc6..481b0459b5f038dfe966ec047646e01c08ed3208 100644 --- a/geomagio/processing/factory.py +++ b/geomagio/processing/factory.py @@ -1,6 +1,5 @@ import os -from ..TimeseriesFactory import TimeseriesFactory from ..edge import EdgeFactory, MiniSeedFactory @@ -9,7 +8,7 @@ def get_edge_factory( host=os.getenv("EDGE_HOST", "127.0.0.1"), interval="second", **kwargs -) -> TimeseriesFactory: +) -> EdgeFactory: return EdgeFactory(host=host, interval=interval, type=data_type, **kwargs) @@ -18,5 +17,5 @@ def get_miniseed_factory( host=os.getenv("EDGE_HOST", "127.0.0.1"), interval="second", **kwargs -) -> TimeseriesFactory: +) -> MiniSeedFactory: return MiniSeedFactory(host=host, interval=interval, type=data_type, **kwargs)