zoriya/Autopipe
1
0
Fork 0
mirror of https://github.com/zoriya/Autopipe.git synced 2026-06-03 23:15:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing the coordinator selection

Browse Source
This commit is contained in:
Zoe Roux
2020-11-08 20:46:09 +01:00
parent 3e2c2846d0
commit 1f0241bb5b
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
autopipe/autopipe.py
+3 -2
View File
@@ -5,6 +5,7 @@ from autopipe import available_coordinators, ArgumentError
class Autopipe:
def __init__(self, coordinator, coordinator_args, log_level=logging.WARNING):
logging.basicConfig(format="%(levelname)s: %(message)s", level=log_level)
if coordinator not in available_coordinators:
coordinator_class = next((i for i in available_coordinators if i.name() == coordinator), None)
if coordinator_class is None:
raise ArgumentError(f"Invalid coordinator: {coordinator}", "coordinator")
self.coordinator = available_coordinators[coordinator](*coordinator_args)
self.coordinator = coordinator_class(*coordinator_args)