refactor

pinky/src/data.py

@@ -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,
+        )