missing import

Author: cgevans

src/riverine/echo.py

@@ -10,7 +10,7 @@
 
 from riverine.util import gen_random_hash, maybe_cache_once
 
-from .actions import AbstractAction, ActionWithComponents
+from .actions import AbstractAction, ActionWithComponents, _STRUCTURE_CLASSES
 from .experiments import Experiment
 from .mixes import Mix
 from .printing import MixLine
@@ -505,7 +505,6 @@ def each_volumes(
             round(maybe_vol)
             * self.droplet_volume
         ]
-        return ea_vols
 
     @maybe_cache_once
     def _mixlines(
@@ -542,3 +541,6 @@ def _mixlines(
 #     """Use an intermediate mix to obtain a target concentration."""
 
 #     ...
+
+for c in [EchoFixedVolume, EchoEqualTargetConcentration, EchoTargetConcentration, EchoFillToVolume]:
+    _STRUCTURE_CLASSES[c.__name__] = c