Skip to content

Commit f9826eb

Browse files
authored
fix typo (#165)
1 parent ea7d2d6 commit f9826eb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

compiler_opt/rl/train_locally.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -137,13 +137,13 @@ def sequence_example_iterator_fn(seq_ex: List[str]):
137137
len(reward_stat_map))
138138

139139
best_trajectory_repo = None
140-
best_trajecroty_repo_path = os.path.join(root_dir,
140+
best_trajectory_repo_path = os.path.join(root_dir,
141141
'best_trajectory_repo.json')
142142
if dump_best_trajectory:
143143
best_trajectory_repo = best_trajectory.BestTrajectoryRepo(
144144
action_name=action_spec.name)
145-
if tf.io.gfile.exists(best_trajecroty_repo_path):
146-
best_trajectory_repo.load_from_json_file(best_trajecroty_repo_path)
145+
if tf.io.gfile.exists(best_trajectory_repo_path):
146+
best_trajectory_repo.load_from_json_file(best_trajectory_repo_path)
147147

148148
with worker_manager_class(
149149
worker_class=problem_config.get_runner_type(),
@@ -168,7 +168,7 @@ def sequence_example_iterator_fn(seq_ex: List[str]):
168168
json.dump(reward_stat_map, f, cls=constant.DataClassJSONEncoder)
169169

170170
if best_trajectory_repo is not None:
171-
best_trajectory_repo.sink_to_json_file(best_trajecroty_repo_path)
171+
best_trajectory_repo.sink_to_json_file(best_trajectory_repo_path)
172172

173173
policy_path = os.path.join(root_dir, 'policy',
174174
str(llvm_trainer.global_step_numpy()))

0 commit comments

Comments
 (0)