Allow overriding of runner mappings in Runner class

NVIDIA · May 15, 2024 · cf566dd · cf566dd
1 parent f3c6951
commit cf566dd
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/cloudai/runner/core/runner.py b/src/cloudai/runner/core/runner.py
@@ -61,8 +61,6 @@ def register(cls, system_type: str) -> Callable:
         """
 
         def decorator(runner_class: Type[BaseRunner]) -> Type[BaseRunner]:
-            if system_type in cls._runners:
-                raise KeyError(f"Runner for {system_type} already registered.")
             cls._runners[system_type] = runner_class
             return runner_class