commit 7a4ea77e8b41f8717174594e048f723ef9875e8e
author yundiqian <yundi@google.com> Mon Oct 10 19:02:55 2022 +0000
committer yundiqian <yundi@google.com> Mon Oct 10 19:02:55 2022 +0000
tree 6cb750c1c4991b992e264f167a341112b2067d87
parent ea7d2d616c036d55b56eaaab3a7157dee3cb3673

fix typo

compiler_opt/rl/train_locally.py

diff --git a/compiler_opt/rl/train_locally.py b/compiler_opt/rl/train_locally.py
index 26526e5..5d819c0 100644
--- a/compiler_opt/rl/train_locally.py
+++ b/compiler_opt/rl/train_locally.py
@@ -137,13 +137,13 @@
                  len(reward_stat_map))
 
   best_trajectory_repo = None
-  best_trajecroty_repo_path = os.path.join(root_dir,
+  best_trajectory_repo_path = os.path.join(root_dir,
                                            'best_trajectory_repo.json')
   if dump_best_trajectory:
     best_trajectory_repo = best_trajectory.BestTrajectoryRepo(
         action_name=action_spec.name)
-    if tf.io.gfile.exists(best_trajecroty_repo_path):
-      best_trajectory_repo.load_from_json_file(best_trajecroty_repo_path)
+    if tf.io.gfile.exists(best_trajectory_repo_path):
+      best_trajectory_repo.load_from_json_file(best_trajectory_repo_path)
 
   with worker_manager_class(
       worker_class=problem_config.get_runner_type(),
@@ -168,7 +168,7 @@
         json.dump(reward_stat_map, f, cls=constant.DataClassJSONEncoder)
 
       if best_trajectory_repo is not None:
-        best_trajectory_repo.sink_to_json_file(best_trajecroty_repo_path)
+        best_trajectory_repo.sink_to_json_file(best_trajectory_repo_path)
 
       policy_path = os.path.join(root_dir, 'policy',
                                  str(llvm_trainer.global_step_numpy()))