simplify imports

Signed-off-by: Jack Luar <[email protected]>

Author: luarss

tools/AutoTuner/distributed/ray_wrapper.py

@@ -1,10 +1,9 @@
 import argparse
 import os
 import datetime
+import yaml
 from ray.job_submission import JobSubmissionClient
 
-from autotuner.utils.config_helper import parse_config
-
 # Constants
 FILE_DIR = os.path.dirname(os.path.abspath(__file__))
 EXPERIMENT_DATETIME = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
@@ -43,7 +42,9 @@
 
 # Parse configs
 # TODO: Validations + Schemas
-config_args = parse_config(os.path.join(FILE_DIR, "config.yaml"))
+print("Parsing config file...")
+with open(os.path.join(FILE_DIR, "config.yaml")) as file:
+    config_args = yaml.safe_load(file)
 
 
 if __name__ == "__main__":