From 2e91e4a5162eb9c2325cc8ad026570eeb7812d2f Mon Sep 17 00:00:00 2001 From: Tom George Date: Mon, 19 Aug 2024 09:25:51 +0100 Subject: [PATCH] minor convention shift of dimension name attributes --- ratinabox/Environment.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ratinabox/Environment.py b/ratinabox/Environment.py index fa7e761..0671aef 100644 --- a/ratinabox/Environment.py +++ b/ratinabox/Environment.py @@ -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()."