From a34dbb0ca174e524cf26c7d13ec70ca281917251 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Tue, 17 Sep 2024 09:59:26 +0200
Subject: [PATCH] Force this stuff.

---
 integration-tests/conftest.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/integration-tests/conftest.py b/integration-tests/conftest.py
index 3130421e..83ba2d71 100644
--- a/integration-tests/conftest.py
+++ b/integration-tests/conftest.py
@@ -369,6 +369,12 @@ def launcher(event_loop):
             args.append("--disable-grammar-support")
         if num_shard is not None:
             args.extend(["--num-shard", str(num_shard)])
+        else:
+            try:
+                os.environ["WORLD_SIZE"] = 1
+                del os.environ["NUM_SHARD"]
+            except Exception:
+                pass
         if quantize is not None:
             args.append("--quantize")
             args.append(quantize)