fix syntax in tests

NSLS-II · Jul 7, 2024 · 388cc3b · 388cc3b
1 parent 4e1db58
commit 388cc3b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/src/blop/agent.py b/src/blop/agent.py
@@ -126,12 +126,11 @@ def __init__(
 
         self.dofs = DOFList(list(np.atleast_1d(dofs)))
         self.objectives = ObjectiveList(list(np.atleast_1d(objectives)))
+        self.detectors = list(np.atleast_1d(detectors))
 
         _validate_dofs_and_objs(self.dofs, self.objectives)
 
         self.db = db
-
-        self.detectors = detectors
         self.acquisition_plan = acquistion_plan
         self.digestion = digestion
         self.digestion_kwargs = digestion_kwargs

diff --git a/src/blop/tests/test_acqfs.py b/src/blop/tests/test_acqfs.py
@@ -7,7 +7,7 @@
 @pytest.mark.parametrize("agent", all_agents, indirect=True)
 def test_analytic_acqfs(agent, RE, db, acqf):
     agent.db = db
-    RE(agent.learn("qr", n=16))
+    RE(agent.learn("qr", n=4))
     RE(agent.learn(acqf, n=1))
     getattr(agent, acqf)
 

diff --git a/src/blop/tests/test_sims.py b/src/blop/tests/test_sims.py
@@ -24,7 +24,7 @@ def test_kb_simulation(RE, db):
     agent = Agent(
         dofs=dofs,
         objectives=objectives,
-        dets=[beamline.det],
+        detectors=beamline.det,
         verbose=True,
         db=db,
         tolerate_acquisition_errors=False,