inp.value == None -> inp.value is None (#1392)

Lightning-AI · Nov 4, 2024 · b74aebf · b74aebf
1 parent 1de5dd8
commit b74aebf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py
@@ -5750,7 +5750,7 @@ def _get_fake_arg(inp: Any):
     if inp is None:
         return inp
     if isinstance(inp, NumberProxy):
-        if inp.value == None:
+        if inp.value is None:
             raise NotImplementedError("Unsupported for NumberProxy.value=None")
         else:
             return inp.value