Minor edits

UrbsLab · May 9, 2024 · f9b0dfc · f9b0dfc
1 parent 2cec466
commit f9b0dfc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/run.py b/run.py
@@ -11,8 +11,7 @@
 if __name__ == '__main__':
     stl_runner = STREAMLINERunner()
     stl_runner.process_argv(sys.argv)
-    stl_runner.set_logger()
+    logger = stl_runner.set_logger()
     # logging.warn(stl_runner.params)
     stl_runner.run()
-    logging.warn("FINISHED!!")
-
+    logging.warning("FINISHED!!")
diff --git a/streamline/runner.py b/streamline/runner.py
@@ -117,6 +117,7 @@ def set_logger(self):
             file_handler.setLevel(logging.INFO)
             file_handler.setFormatter(formatter)
             logger.addHandler(file_handler)
+        return logger
 
     def get_len_datasets(self):
         output_path, experiment_name  = self.params['output_path'], self.params['experiment_name']