factorize create_all_type_evaluators

Drakkar-Software · Jan 12, 2020 · 3898127 · 3898127
1 parent ee904c9
commit 3898127
Showing 1 changed file with 3 additions and 23 deletions.
diff --git a/octobot_evaluators/api/evaluators.py b/octobot_evaluators/api/evaluators.py
@@ -133,26 +133,6 @@ async def create_all_type_evaluators(config, exchange_name,
                                      symbols=None,
                                      time_frames=None,
                                      relevant_evaluators=CONFIG_WILDCARD) -> list:
-    created_evaluators = []
-
-    created_evaluators += await create_evaluators(EvaluatorClassTypes[EvaluatorMatrixTypes.TA.value],
-                                                  config, exchange_name, symbols=symbols, time_frames=time_frames,
-                                                  cryptocurrencies=cryptocurrencies,
-                                                  relevant_evaluators=relevant_evaluators)
-
-    created_evaluators += await create_evaluators(EvaluatorClassTypes[EvaluatorMatrixTypes.SOCIAL.value],
-                                                  config, exchange_name, symbols=symbols, time_frames=time_frames,
-                                                  cryptocurrencies=cryptocurrencies,
-                                                  relevant_evaluators=relevant_evaluators)
-
-    created_evaluators += await create_evaluators(EvaluatorClassTypes[EvaluatorMatrixTypes.REAL_TIME.value],
-                                                  config, exchange_name, symbols=symbols, time_frames=time_frames,
-                                                  cryptocurrencies=cryptocurrencies,
-                                                  relevant_evaluators=relevant_evaluators)
-
-    created_evaluators += await create_evaluators(EvaluatorClassTypes[EvaluatorMatrixTypes.STRATEGIES.value],
-                                                  config, exchange_name, symbols=symbols, time_frames=time_frames,
-                                                  cryptocurrencies=cryptocurrencies,
-                                                  relevant_evaluators=relevant_evaluators)
-
-    return created_evaluators
+    return [await create_evaluators(evaluator_type, config, exchange_name, symbols=symbols, time_frames=time_frames,
+                                    cryptocurrencies=cryptocurrencies, relevant_evaluators=relevant_evaluators)
+            for evaluator_type in EvaluatorClassTypes.values()]