AVSLab / bsk_rl

RL environments and tools for spacecraft autonomy research, built on Basilisk. Developed by the AVS Lab.
https://avslab.github.io/bsk_rl/
MIT License
39 stars 4 forks source link

Issue #58: Expose state vector element names #111

Closed Mark2000 closed 8 months ago

Mark2000 commented 8 months ago

Description

Closes #58

Allows the user to print out a list of state names from sat observations. Can only be produced after environment initialization, since the shape of each observation element is unknown until then. For example

Observation: [ 5.55643668e-04 -8.20527293e-04 -2.82089986e-03  1.30471109e-03
              -2.84886806e-05  9.99999148e-01  5.31311297e-01 -5.76592276e-01
              -7.38774349e-01  5.67862249e-01  7.45917635e-01 -1.73853876e-01
               9.84419538e-01  4.82351662e-01  7.87235656e-01  2.62243219e-01
              -5.58111562e-01  6.60252514e-01  7.85183742e-01  2.64519689e-01
              -5.59924839e-01  4.17639105e-01  7.78825097e-01  2.71977900e-01
              -5.65207475e-01  3.44327485e-01]
States: ['omega_BP_P_normd[0]', 'omega_BP_P_normd[1]', 'omega_BP_P_normd[2]', 
         'c_hat_P[0]', 'c_hat_P[1]', 'c_hat_P[2]', 'r_BN_P_normd[0]', 'r_BN_P_normd[1]', 
         'r_BN_P_normd[2]', 'v_BN_P_normd[0]', 'v_BN_P_normd[1]', 'v_BN_P_normd[2]', 
         'battery_charge_fraction', 'target_obs.target_0.priority', 
         'target_obs.target_0.location_normd[0]',  'target_obs.target_0.location_normd[1]', 
         'target_obs.target_0.location_normd[2]',  'target_obs.target_1.priority', 
         'target_obs.target_1.location_normd[0]', 'target_obs.target_1.location_normd[1]', 
         'target_obs.target_1.location_normd[2]', 'target_obs.target_2.priority', 
         'target_obs.target_2.location_normd[0]', 'target_obs.target_2.location_normd[1]',
         'target_obs.target_2.location_normd[2]', 'normalized_time']

Type of change

How should this pull request be reviewed?

How Has This Been Tested?

Please describe the tests that you ran to verify your changes.

Passes Tests

Test Configuration

Checklist: