outputResultJSON of Operator does not update entity of schedule if entity is not instance of Job

dream/simulation/Operator.py

@@ -303,6 +303,8 @@ class Operator(ObjectResource):
                 if record.get("exitTime", None) != None:
                     json['results']['schedule'][-1]["exitTime"] = record["exitTime"]
                 if record.get("entity", None):
+                    from Job import Job
+                    if issubclass(record["entity"].__class__, Job):
                         json['results']['schedule'][-1]["entityId"] = record["entity"].id
                 if record.get("task_id", None):
                     json['results']['schedule'][-1]["task_id"] = record["task_id"]

dream/tests/dump/SkilledOperator02.json.result

@@ -83,13 +83,11 @@
     ], 
     "schedule": [
      {
-      "entityId": "P1", 
       "entranceTime": 0, 
       "exitTime": 17.0, 
       "stationId": "M1"
      }, 
      {
-      "entityId": "P1", 
       "entranceTime": 20.0, 
       "exitTime": 22.0, 
       "stationId": "M2"

dream/tests/dump/SkilledOperatorParallel01.json.result

@@ -92,8 +92,7 @@
     ], 
     "schedule": [
      {
-      "entityId": "Batch22047", 
-      "entranceTime": 0, 
+      "entranceTime": 0.0, 
       "stationId": "M1"
      }
     ], 
@@ -115,13 +114,11 @@
     ], 
     "schedule": [
      {
-      "entityId": "Batch22048", 
       "entranceTime": 0, 
       "exitTime": 80.0, 
       "stationId": "M2"
      }, 
      {
-      "entityId": "Batch22049", 
       "entranceTime": 80.0, 
       "stationId": "M2"
      }