Gen-Sim commited on
Commit
0c345e1
·
1 Parent(s): f82fd47

Update gensim/sim_runner.py

Browse files
Files changed (1) hide show
  1. gensim/sim_runner.py +1 -1
gensim/sim_runner.py CHANGED
@@ -91,7 +91,7 @@ class SimulationRunner:
91
  if os.path.exists("cliport/generated_tasks/" + task_file):
92
  online_code_buffer[task_file] = open("cliport/generated_tasks/" + task_file).read()
93
 
94
- random_task_file = random.sample(list(online_code_buffer.keys()), 1)
95
  class_def = [line for line in online_code_buffer[task_file].split("\n") if line.startswith('class')]
96
  task_name = class_def[0]
97
  task_name = task_name[task_name.find("class "): task_name.rfind("(Task)")][6:]
 
91
  if os.path.exists("cliport/generated_tasks/" + task_file):
92
  online_code_buffer[task_file] = open("cliport/generated_tasks/" + task_file).read()
93
 
94
+ random_task_file = random.sample(list(online_code_buffer.keys()), 1)[0]
95
  class_def = [line for line in online_code_buffer[task_file].split("\n") if line.startswith('class')]
96
  task_name = class_def[0]
97
  task_name = task_name[task_name.find("class "): task_name.rfind("(Task)")][6:]