fix(DataStreamAgent): insert missing legacy import for agents.DataStr…

…eamAgent
Met4FoF · Aug 27, 2021 · 302f61a · 302f61a
1 parent 16d8e56
commit 302f61a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/agentMET4FOF/agents/__init__.py b/agentMET4FOF/agents/__init__.py
@@ -1,5 +1,5 @@
 """This package provides access to all agents included in agentMET4FOF"""
-from .base_agents import AgentMET4FOF, MonitorAgent
+from .base_agents import AgentMET4FOF, DataStreamAgent, MonitorAgent
 from .metrological_base_agents import (
     MetrologicalAgent,
     MetrologicalMonitorAgent,
@@ -13,6 +13,7 @@
     "AgentBuffer",
     "AgentMET4FOF",
     "AgentNetwork",
+    "DataStreamAgent",
     "MetrologicalAgent",
     "MetrologicalAgentBuffer",
     "MetrologicalMonitorAgent",

diff --git a/tests/test_all_imports.py b/tests/test_all_imports.py
@@ -4,6 +4,7 @@ def test_legacy_imports():
         AgentBuffer,
         AgentMET4FOF,
         AgentNetwork,
+        DataStreamAgent,
         MonitorAgent,
         SineGeneratorAgent,
     )