Errors in examples

[andrewczgithub]

Hi @apoorva-sharma @blandry @BorisIvanovic
cheers really appreciate the help.
Just a few queries. Im getting an error with imports on the example

multistep-varying-noise.ipynb

ModuleNotFoundError Traceback (most recent call last)
in
3 import numpy as np
4 import gym
----> 5 from main.maml_agent import MAMLAgent, register_flags
6 from main.alpaca import *
7 from main.dataset import *

~/ALPaCA/main/maml_agent.py in
4 from maml.data_generator import DataGenerator
5 from tensorflow.python.platform import flags
----> 6 from main.datagen import *
7 from main.dataViz import *
8 import yaml

ModuleNotFoundError: No module named 'main.datagen'

Also how do I set up the config file for an external dataset?

Many thanks,
Best,
Andrew