From 8aadba04f5c6d5ef52cfc03829bb786d55f7d8c4 Mon Sep 17 00:00:00 2001
From: gkroiz <gersonkroiz@google.com>
Date: Mon, 8 Apr 2024 23:27:51 +0000
Subject: [PATCH] minor typo fix

---
 src/litdata/utilities/env.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/litdata/utilities/env.py b/src/litdata/utilities/env.py
index 874e597b..7d90d4de 100644
--- a/src/litdata/utilities/env.py
+++ b/src/litdata/utilities/env.py
@@ -66,7 +66,7 @@ def detect(cls) -> "_DistributedEnv":
         if os.environ["NNODES"] is not None:
             num_nodes = os.environ["NNODES"]
 
-        if world_size is in [None, -1, 0]:
+        if world_size in (None, -1, 0):
             world_size = 1
 
         return cls(world_size=world_size, global_rank=global_rank, num_nodes=num_nodes)