cron test error -- ./modules/transforms_demo_2d.ipynb

[wyli]

Describe the bug
logs from https://github.com/Project-MONAI/MONAI/runs/3604008777?check_suite_focus=true

real	0m10.738s
user	0m15.965s
sys	0m10.666s
Running ./modules/transforms_demo_2d.ipynb
Checking PEP8 compliance...
Running notebook...

Executing:   0%|          | 0/20 [00:00<?, ?cell/s]
Executing:   5%|▌         | 1/20 [00:00<00:15,  1.25cell/s]
Executing:  15%|█▌        | 3/20 [00:05<00:32,  1.92s/cell]
Executing:  20%|██        | 4/20 [00:07<00:30,  1.91s/cell]
Executing:  50%|█████     | 10/20 [00:25<00:28,  2.81s/cell]
Executing:  60%|██████    | 12/20 [00:26<00:16,  2.12s/cell]
Executing:  70%|███████   | 14/20 [00:29<00:12,  2.01s/cell]
Executing:  75%|███████▌  | 15/20 [00:30<00:08,  1.75s/cell]
Executing:  75%|███████▌  | 15/20 [00:31<00:10,  2.08s/cell]
Traceback (most recent call last):
  File "/opt/conda/bin/papermill", line 8, in <module>
    sys.exit(papermill())
  File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 829, in __call__
    return self.main(*args, **kwargs)
  File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 782, in main
    rv = self.invoke(ctx)
  File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 1066, in invoke
    return ctx.invoke(self.callback, **ctx.params)
  File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 610, in invoke
    return callback(*args, **kwargs)
  File "/opt/conda/lib/python3.8/site-packages/click/decorators.py", line 21, in new_func
    return f(get_current_context(), *args, **kwargs)
  File "/opt/conda/lib/python3.8/site-packages/papermill/cli.py", line 250, in papermill
    execute_notebook(
  File "/opt/conda/lib/python3.8/site-packages/papermill/execute.py", line 122, in execute_notebook
    raise_for_execution_errors(nb, output_path)
  File "/opt/conda/lib/python3.8/site-packages/papermill/execute.py", line 234, in raise_for_execution_errors
    raise error
papermill.exceptions.PapermillExecutionError: 
---------------------------------------------------------------------------
Exception encountered at "In [9]":
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/tmp/ipykernel_45104/2592799331.py in <module>
      2 plt.subplot(1, 2, 1)
      3 plt.title("image")
----> 4 plt.imshow(np.moveaxis(new_img.astype(int), 0, -1))
      5 plt.subplot(1, 2, 2)
      6 plt.title("label")

AttributeError: 'Tensor' object has no attribute 'astype'