DebasishDhal99 commited on
Commit
fcf43ac
1 Parent(s): 5e24ac5

Update multi_agent_2D.py

Browse files
Files changed (1) hide show
  1. multi_agent_2D.py +3 -1
multi_agent_2D.py CHANGED
@@ -124,9 +124,11 @@ def multi_agent_walk(agent_count, iters, step_size = 1, random_seed = None):
124
 
125
  fig.canvas.draw()
126
  image_array = np.array(fig.canvas.renderer.buffer_rgba())
 
 
127
 
128
  try:
129
- return image_array, df
130
  except:
131
  return image_array, None
132
 
 
124
 
125
  fig.canvas.draw()
126
  image_array = np.array(fig.canvas.renderer.buffer_rgba())
127
+ csv_file = "2d_random_walk_coordinates.csv"
128
+ df.to_csv(csv_file, index=False)
129
 
130
  try:
131
+ return image_array, csv_file
132
  except:
133
  return image_array, None
134