SE-2289 added back compatibility for DNNStatus

labforge · Jun 2, 2023 · 44ef673 · 44ef673
1 parent 1970df5
commit 44ef673
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/utility/uploader.py b/utility/uploader.py
@@ -43,7 +43,10 @@ def __init__(self, address, flag_name, status_name, device, filename, parent=Non
         self.calibrate = not self.filename.lower().endswith(".tar")
         if not self.calibrate:
             self.flag = self.device.GetParameters().Get(flag_name)
-            self.status = self.device.GetParameters().Get(status_name)        
+            for name in status_name:
+                self.status = self.device.GetParameters().Get(name)  
+                if self.status is not None:                    
+                    break       
 
             if flag_name.lower().find('weight') > 0:
                 self.reset_flag = True

diff --git a/utility/utility.py b/utility/utility.py
@@ -240,10 +240,10 @@ def handle_upload(self):
 
         if ftype == "Firmware":
             update_flag = "EnableUpdate"
-            update_status = "UpdateStatus"
+            update_status = ["UpdateStatus"]
         elif ftype == "DNN Weights":
             update_flag = "EnableWeightsUpdate"
-            update_status = "WeightsStatus"
+            update_status = ["WeightsStatus", "DNNStatus"]
         elif ftype == "Calibration":
             update_flag = None
             update_status = None