Commit c1f59c33 authored by Marius Kriegerowski's avatar Marius Kriegerowski

refactor

parent 2dbb3e55
......@@ -302,7 +302,7 @@ class GFSwarmData(DataGenerator):
return (n, e, source.depth)
def generate(self):
sources = swarm.get_effective_sources()
sources = self.swarm.get_effective_sources()
self.tensor_shape = (len(self.targets), self.n_samples_max)
response = self.swarm.engine.process(
......@@ -322,8 +322,16 @@ class GFSwarmData(DataGenerator):
gf_engine = LocalEngine(
use_config=True,
store_superdirs=['/data/stores'],
default_store_id='vogtland_001')
return cls(fn_stations='stations.pf', gf_engine=gf_engine,
sample_length=10)
# default_store_id='test_store'
)
example_swarm = source_region.Swarm(
geometry=source_region.CuboidSourceGeometry(),
timing=source_region.RandomTiming(),
engine=gf_engine
)
return cls(
fn_stations='stations.pf', sample_length=10, swarm=example_swarm,
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment