diff --git a/prepare_yaml_file.py b/prepare_yaml_file.py
index 12dae8c5..2be0451d 100644
--- a/prepare_yaml_file.py
+++ b/prepare_yaml_file.py
@@ -1,11 +1,5 @@
-import os
 import sys
-import time
 import yaml
-import json
-import pynvml 
-import time
-import psutil
 
 def main():
     run_type = sys.argv[1]