refactor the method 'file_sink' of the packages 'ml/utils'

tdspora · Oct 31, 2024 · 64ee071 · 64ee071
1 parent ad8400e
commit 64ee071
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/syngen/VERSION b/src/syngen/VERSION
@@ -1 +1 @@
-0.9.49
+0.9.50rc0
diff --git a/src/syngen/ml/utils/utils.py b/src/syngen/ml/utils/utils.py
@@ -366,9 +366,11 @@ def file_sink(message):
     """
     Save logs to the log file
     """
-    with open(os.getenv("SUCCESS_LOG_FILE"), "a") as log_file:
-        log_message = fetch_log_message(message)
-        log_file.write(log_message + "\n")
+    path_to_logs = os.getenv("SUCCESS_LOG_FILE")
+    if path_to_logs is not None:
+        with open(path_to_logs, "a") as log_file:
+            log_message = fetch_log_message(message)
+            log_file.write(log_message + "\n")
 
 
 def console_sink(record):