Depth save

allenai · Aug 22, 2022 · 4fc5795 · 4fc5795
1 parent 65d76f9
commit 4fc5795
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ai2thor/tests/test_unity.py b/ai2thor/tests/test_unity.py
@@ -189,7 +189,7 @@ def depth_images_near(depth1, depth2, epsilon=1e-5, debug_save=False, filepath="
         max = np.max(diff)
         norm_diff = diff / max
         dx = np.where(np.abs(depth1 - depth2) >= epsilon)
-        depth_copy[dx] = norm_diff[dx] * (255, 0, 255)
+        depth_copy[dx] =  (norm_diff[dx]*255, norm_diff[dx] * 0, norm_diff[dx] *255)
         test_name = os.environ.get('PYTEST_CURRENT_TEST').split(':')[-1].split(' ')[0]
         debug_directory = os.path.join(os.path.join(os.getcwd(), TEST_OUTPUT_DIRECTORY))
         # if os.path.exists(debug_directory):