Skip to content

Commit

Permalink
minor convention shift of dimension name attributes
Browse files Browse the repository at this point in the history
  • Loading branch information
TomGeorge1234 committed Aug 19, 2024
1 parent a59f9c5 commit 2e91e4a
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions ratinabox/Environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,13 +193,14 @@ def __init__(self, params={}):
if self.D == 1:
self.dimensions = {'x' : self.discrete_coords[:,0]}
self.dim_names = ['x']
self.rate_map_dim_order = ['x']
elif self.D == 2:
self.dimensions = {
'y' : self.discrete_coords[:,0,1],
'x' : self.discrete_coords[0,:,0],
}
self.dim_names = ['y','x'] # this is ordered according to the order of the dimensions in the discrete_coords array (y then x)

self.dim_names = ['x','y'] # this is ordering of the dimensions in, for example, the agents position
self.rate_map_dim_order = ['y','x'] # this is the order of the dimensions in rate maps (y then x)
if ratinabox.verbose is True:
print(
f"\nAn Environment has been initialised with parameters: {self.params}. Use Env.add_wall() to add a wall into the Environment. Plot Environment using Env.plot_environment()."
Expand Down

0 comments on commit 2e91e4a

Please sign in to comment.