Some trivial code cleanup in run_rai.py (#57)

Azure · Apr 27, 2022 · 0fa8346 · 0fa8346
1 parent c5b147b
commit 0fa8346
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/single-rai-job/src/run_rai.py b/single-rai-job/src/run_rai.py
@@ -402,14 +402,11 @@ def main(args):
     _logger.info("Triggering computation")
     rai_i.compute()
 
-    _logger.info("Doing local save")
+    _logger.info("Doing local save and upload")
     binary_dir = "./responsibleai"
-    rai_i.save("./responsibleai")
+    rai_i.save(binary_dir)
     my_run.upload_folder("dashboard", binary_dir)
 
-    # _logger.info("Saving binary output")
-    # rai_i.save(args.dashboard)
-
     _logger.info("Saving UX JSON")
     rai_data = rai_i.get_data()
     rai_dict = serialize_json_safe(rai_data)