Fix numba-cuda installation for Python 3.9 (#6097)

- The latest numba-cuda release (0.21) is not compatible with Python 3.9.
- This PR restricts the numba-cuda version to 0.20.x for Python 3.9,
  while allowing newer versions for other Python versions.

Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>

Author: JanuszL

qa/TL0_python-self-test-operators_1/test_numba_cuda.sh

@@ -1,6 +1,6 @@
 #!/bin/bash -e
 # used pip packages
-pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa scipy nvidia-ml-py==11.450.51 numba lz4 numba_cuda[cu${DALI_CUDA_MAJOR_VERSION}]>0.19.0'
+pip_packages='${python_test_runner_package} dataclasses numpy opencv-python pillow librosa scipy nvidia-ml-py==11.450.51 numba lz4 numba-cuda'
 
 target_dir=./dali/test/python
 

qa/setup_packages.py

@@ -132,7 +132,7 @@ def get_name(self, cuda_version=None, idx=None):
         cuda_version : str, optional, default = None
             Cuda version used for this query
         idx : int
-            Index of name to retrive in case of specific version has different alias
+            Index of name to retrieve in case of specific version has different alias
         """
         name = BasePackage.get_alias(self.get_version(idx, cuda_version))
         if name is None:
@@ -181,7 +181,11 @@ def get_version(self, idx, cuda_version=None):
         if idx is None:
             idx = 0
         idx = self.clamp_index(idx, cuda_version)
-        return self.get_all_versions(cuda_version)[idx]
+        versions = self.get_all_versions(cuda_version)
+        if len(versions):
+            return versions[idx]
+        else:
+            return None
 
     def get_all_versions(self, cuda_version=None):
         """Get all versions compatible with provided cuda_version
@@ -579,6 +583,19 @@ def get_pyvers_name(self, url, cuda_version):
             ]
         },
     ),
+    CudaPackage(
+        "numba-cuda",
+        {
+            "120": [
+                PckgVer(
+                    "0.20.1", python_min_ver="3.9", python_max_ver="3.9", dependencies=["numpy<2"]
+                ),
+                PckgVer("0.21.1", python_min_ver="3.10", dependencies=["numpy<2"]),
+            ]
+        },
+        # name used during installation
+        name="numba-cuda[cu{cuda_v[0]}{cuda_v[1]}]",
+    ),
 ]
 
 all_packages_keys = [pckg.key for pckg in all_packages]