Coverage for biobb_pytorch / mdae / utils / log_utils.py: 64%

1import os 

2 

3 

4def get_size(path): 

5 size = os.path.getsize(path) 

6 if size < 1024: 

7 return f"{size} bytes" 

8 elif size < pow(1024, 2): 

9 return f"{round(size / 1024, 2)} KB" 

10 elif size < pow(1024, 3): 

11 return f"{round(size / (pow(1024, 2)), 2)} MB" 

12 elif size < pow(1024, 4): 

13 return f"{round(size / (pow(1024, 3)), 2)} GB"