nightly test error -- ./modules/postprocessing_transforms.ipynb

[wyli]

Describe the bug
example log from the tutorial repo, but this is potentially due to a core codebase issue:

real	0m50.018s
user	1m39.868s
sys	0m21.127s
Running ./modules/postprocessing_transforms.ipynb
Checking PEP8 compliance...
Running notebook...
Before:
    "max_epochs = 160\n",
After:
    "max_epochs = 1\n",
Before:
    "val_interval = 2\n",
After:
    "val_interval = 1\n",

Executing:   0%|          | 0/26 [00:00<?, ?cell/s]
Executing:   4%|▍         | 1/26 [00:00<00:24,  1.02cell/s]
Executing:  12%|█▏        | 3/26 [00:05<00:32,  1.43s/cell]
Executing:  15%|█▌        | 4/26 [00:08<00:40,  1.82s/cell]
Executing:  38%|███▊      | 10/26 [01:00<01:01,  3.87s/cell]
Executing:  69%|██████▉   | 18/26 [01:56<00:38,  4.80s/cell]
Executing:  77%|███████▋  | 20/26 [01:58<00:21,  3.64s/cell]
Executing:  85%|████████▍ | 22/26 [01:58<00:10,  2.57s/cell]
Executing:  85%|████████▍ | 22/26 [02:00<00:21,  5.47s/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 [11]":
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-11-a0b3f1d6c2a2> in <module>
      6 metric_values = []
      7 
----> 8 post_pred = Compose([EnsureType(), AsDiscrete(argmax=True, to_onehot=True, num_classes=2)])
      9 post_label = Compose([EnsureType(), AsDiscrete(to_onehot=True, num_classes=2)])
     10 

TypeError: __init__() got an unexpected keyword argument 'num_classes'