Merge pull request #234 from zhujun98/fix_bug_in_special_suite_pipeline

Fix correlate bug in special suite
European-XFEL · Jun 22, 2020 · 1598f3b · 1598f3b
2 parents 3cb1b48 + 5e2dc08
commit 1598f3b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/extra_foam/special_suite/special_analysis_base.py b/extra_foam/special_suite/special_analysis_base.py
@@ -642,7 +642,7 @@ def run(self):
             "timeout": config["CLIENT_TIME_OUT"]
         }
         if self._client_instance_type == KaraboGateClient:
-            kwargs["connect"] =  False
+            kwargs["connect"] = False
 
         with self._client_instance_type(self._endpoint_st, **kwargs) as client:
             self.log.info(f"Connected to {self._endpoint_st}")
@@ -699,7 +699,7 @@ def run(self):
                     continue
 
                 try:
-                    correlated, dropped = self._transformer_st.correlate(data)
+                    correlated, _, dropped = self._transformer_st.correlate(data)
                     for tid, err in dropped:
                         self.log.error(err)
                 except Exception as e: