Fixes

Signed-off-by: Joaquin Anton Guirao <[email protected]>

Author: jantonguirao

dali/python/nvidia/dali/experimental/dali2/_device.py

@@ -25,7 +25,8 @@ def __init__(self, name: str, device_id: int = None):
             raise ValueError(f"Invalid device name: {name}\n"
                              f"Ordinal ':{name_device_id}' should not appear "
                              "in device name when device_id is provided")
-        device_id = device_id or name_device_id
+        if device_id is None:
+            device_id = name_device_id
 
         if device_type == "cuda":
             device_type = "gpu"