UnboundedLocalError: local variable referenced before assignment.

[pranavkantgaur]

Hi,

In test.py, specifically, here, I find that the variable data_shape1ax has been referenced, but if I am using the single network (i.e., if(config_net1) is True) for whole tumor (similarly for other two tumor networks) data_shape1ax (similarly for data_shape2ax and data_shape3ax) will be undefined.

I get the following error:
UnboundedLocalError: local variable 'data_shape1ax' referenced before assignment.