Simplify a bit

dfilan · dfilan · commit 36c7ee4c643c · 2022-12-14T15:46:31.000-08:00
diff --git a/src/reward_preprocessing/interpret.py b/src/reward_preprocessing/interpret.py
@@ -367,8 +367,7 @@ def interpret(
             )
 
             # remove opacity channel from dataset thumbnails
-            num_channels_total = dataset_thumbnails.shape[0]
-            np_trans_tens = dataset_thumbnails[0 : num_channels_total - 1, :, :]
+            np_trans_tens = dataset_thumbnails[:-1, :, :]
 
             obs, _, next_obs = ndarray_to_transition(np_trans_tens)
 

Original file line number	Diff line number	Diff line change
`@@ -367,8 +367,7 @@ def interpret(`
`367`	`367`	`)`
`368`	`368`
`369`	`369`	`# remove opacity channel from dataset thumbnails`
`370`		`- num_channels_total = dataset_thumbnails.shape[0]`
`371`		`- np_trans_tens = dataset_thumbnails[0 : num_channels_total - 1, :, :]`
	`370`	`+ np_trans_tens = dataset_thumbnails[:-1, :, :]`
`372`	`371`
`373`	`372`	`obs, _, next_obs = ndarray_to_transition(np_trans_tens)`
`374`	`373`