tests: Add new pi tag to benchmark Py script.

carlosperate · Jan 2, 2024 · 5a19ecb · 5a19ecb
1 parent d150dbc
commit 5a19ecb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmark-vm.py b/benchmark-vm.py
@@ -134,7 +134,7 @@ def main():
     parser.add_argument("-p", "--pi-version", type=str, default=PI_VERSION, help="(Optional) Raspberry Pi version (pi1, pi2, pi3)")
     args = parser.parse_args()
 
-    if args.pi_version not in ("pi1", "pi2", "pi3"):
+    if args.pi_version not in ("pi1", "pi2", "pi3", "pivirt"):
         raise Exception("Invalid Raspberry Pi version provided: {}".format(args.pi_version))
 
     run(args.docker_img, args.os_img, args.pi_version)