Tensorboard Callback for Fastai

Thanks a lot @Pendar for having such an awesome vision! I wanted to give it a try with a very minimalist example (resnet18 on MNIST running on Sagemaker). For some reason I get the classical error " Failed to load the set of active dashboards." but my back-end is indeed running. I pretty much followed the documentation. Could somebody take a look at the code and tell me what obvious thing I am missing?

!pip install tensorboard
!pip install tensorboardx

from fastai.vision import *
from fastai.callbacks.tensorboard import *
path = untar_data(URLs.MNIST_SAMPLE)
data = ImageDataBunch.from_folder(path)
project_id = 'project1'
tboard_path = Path('data/tensorboard/' + project_id)
learn.callback_fns.append(partial(LearnerTensorboardWriter, 
                                    base_dir=tboard_path, 
                                    name='run1'))
learn.fit_one_cycle(2)
!tensorboard --logdir=data/tensorboard/project1 --port=6006

In my case, localhost should be https://pytorch-tensorboard.notebook.eu-west-1.sagemaker.aws/proxy/6006 according to this.

[UPDATE]: I tried out the sample tutorial (colab data) without any changes, same issue