[NbConvertApp] Converting notebook HCP_downstream_raw_flatmaps.ipynb to python
[NbConvertApp] Writing 36336 bytes to HCP_downstream_raw_flatmaps.py
Traceback (most recent call last):
  File "/weka/proj-fmri/ckadirt/fMRI-foundation-model/src/HCP_downstream_raw_flatmaps.py", line 727, in <module>
    num_iterations_per_epoch = math.ceil(flatmaps_train/batch_size)
                                         ~~~~~~~~~~~~~~^~~~~~~~~~~
TypeError: unsupported operand type(s) for /: 'Dataset' and 'int'