fix(ci): updated the docker rmi command in run_tests.py

sroycho4 · Dec 19, 2023 · 6edf6d8 · 6edf6d8
1 parent 664b961
commit 6edf6d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/run_tests/run_tests.py b/scripts/run_tests/run_tests.py
@@ -240,7 +240,7 @@
 
     # delete pulled image before terminating
     if device == "gpu":
-        os.system("docker rmi $(docker images | grep 'unifyai/ivy:latest-gpu')")
+        os.system("docker rmi unifyai/ivy:latest-gpu")
 
     # if any tests fail, the workflow fails
     if failed: