Fix nightly test errors #476

Job	Run time
get-test-groups	3s
"python=3.8", group_cpu_001	26m 0s
"python=3.8", group_notebooks_cpu_001	21m 36s
"python=3.8", group_spark_001	25m 7s
"python=3.8", group_notebooks_spark_001	23m 35s
"python=3.8", group_notebooks_spark_002	22m 22s
"python=3.8", group_gpu_001	32m 10s
"python=3.8", group_gpu_002	34m 39s
"python=3.8", group_notebooks_gpu_001	35m 47s
"python=3.8", group_notebooks_gpu_002	34m 27s
"python=3.9", group_cpu_001	23m 51s
"python=3.9", group_notebooks_cpu_001	22m 27s
"python=3.9", group_spark_001	23m 24s
"python=3.9", group_notebooks_spark_001	24m 23s
"python=3.9", group_notebooks_spark_002	22m 21s
"python=3.9", group_gpu_001	33m 8s
"python=3.9", group_gpu_002	34m 54s
"python=3.9", group_notebooks_gpu_001	36m 8s
"python=3.9", group_notebooks_gpu_002	34m 39s
	8h 31m 1s

Provide feedback