Added support for loading classes by providing a class reference.

c4aad1c6 · Josué A · 3de3a25b · c4aad1c6
Commit c4aad1c6 authored Jul 07, 2015 by Josué A
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

simso/core/Scheduler.py
+3 -0

No files found.
--- a/simso/core/Scheduler.py
+++ b/simso/core/Scheduler.py
@@ -53,6 +53,9 @@ class SchedulerInfo(object):
        try:
            clas = None
            if self.clas:
+                if type(self.clas) is type:
+                    clas = self.clas
+                else:
                    name = self.clas.rsplit('.', 1)[1]
                    importlib.import_module(self.clas)
                    clas = getattr(importlib.import_module(self.clas), name)