From c89d2ecd18d59d0de16942cb0818ab10db58abbc Mon Sep 17 00:00:00 2001
From: Rasmus Ringdahl <rasmus.ringdahl@liu.se>
Date: Wed, 22 Jan 2025 14:17:31 +0100
Subject: [PATCH] refactor: change to logging framework
---
1_single_core_job/single_core_task.py | 15 ++++++++++-----
1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/1_single_core_job/single_core_task.py b/1_single_core_job/single_core_task.py
index eb3087e..9e8e26d 100644
--- a/1_single_core_job/single_core_task.py
+++ b/1_single_core_job/single_core_task.py
@@ -1,6 +1,10 @@
from datetime import datetime
-import time
+
+import logging
import os
+import time
+
+logger = logging.getLogger(__name__)
def main():
# Read environment variables.
@@ -13,12 +17,12 @@ def main():
# This represents the calculations
current_time = datetime.now()
sleep_time = 60 - current_time.second
- print('{} - Sleeping for {} seconds.'.format(current_time.strftime('%Y-%m-%d %H:%M:%S'), sleep_time))
+ logger.info('%s - Sleeping for %d seconds.',current_time.strftime('%Y-%m-%d %H:%M:%S'), sleep_time)
time.sleep(sleep_time)
# Printing some things to standard output.
- print('\nJob ID:\t\t\t{name}\nJob name:\t\t{id}\nAllocated cores:\t{cores}\nAllocated memory:\t{mem}'.format(
- id=JOB_ID, name=JOB_NAME, cores=NUMBER_OF_CORES,mem=MAXIMUM_MEMORY))
+ logger.info('\nJob ID:\t\t\t%s\nJob name:\t\t%s\nAllocated cores:\t%s\nAllocated memory:\t%s',
+ JOB_ID, JOB_NAME, NUMBER_OF_CORES,MAXIMUM_MEMORY)
# Writing some output to a file based on the Slurm job id.
output_file = '{}.txt'.format(JOB_ID)
@@ -26,7 +30,8 @@ def main():
file.write('This file was created by the job {} with id {}\n'.format
(JOB_NAME, JOB_ID))
- print('\nJob completed.')
+ logger.info('Job completed.')
if __name__ == '__main__':
+ logging.basicConfig(level=logging.INFO)
main()
--
GitLab