koalazf99 commited on
Commit
dd783be
1 Parent(s): 0aa0bab

minor fix: repeat fp substitution

Browse files
Files changed (1) hide show
  1. llm_src/utils/fp_substitution.py +4 -1
llm_src/utils/fp_substitution.py CHANGED
@@ -29,8 +29,11 @@ def generate_fp_nums_multiarith(orig_nums, repeat_time):
29
 
30
  fp_repl_numbers_list = []
31
  num_repl = len(orig_nums)
 
 
 
32
  for num_var in range(repeat_time):
33
- idx_candi = random.randint(0, len(nums_candis) - 1)
34
  num_pick = sorted(nums_candis[idx_candi])
35
  num_order = np.argsort(orig_nums).tolist()
36
  cur_rdx = 0
 
29
 
30
  fp_repl_numbers_list = []
31
  num_repl = len(orig_nums)
32
+ idx_candis = [_ for _ in range(len(nums_candis))]
33
+ random.shuffle(idx_candis)
34
+
35
  for num_var in range(repeat_time):
36
+ idx_candi = idx_candis[num_var]
37
  num_pick = sorted(nums_candis[idx_candi])
38
  num_order = np.argsort(orig_nums).tolist()
39
  cur_rdx = 0