Merge pull request #1602 from dorado-ai-devops/fix-transform-matrix-keyerror

Fix KeyError in run.py when transform_matrix_start is missing

Author: Tom94

scripts/run.py

@@ -282,7 +282,14 @@ def get_scene(scene):
 		print(args.screenshot_frames)
 		for idx in args.screenshot_frames:
 			f = ref_transforms["frames"][int(idx)]
-			cam_matrix = f.get("transform_matrix", f["transform_matrix_start"])
+
+			if 'transform_matrix' in f:
+				cam_matrix = f['transform_matrix']
+			elif 'transform_matrix_start' in f:
+				cam_matrix = f['transform_matrix_start']
+			else:
+				raise KeyError("Missing both 'transform_matrix' and 'transform_matrix_start'")
+			
 			testbed.set_nerf_camera_matrix(np.matrix(cam_matrix)[:-1,:])
 			outname = os.path.join(args.screenshot_dir, os.path.basename(f["file_path"]))