# Import scripts
from src.utils import utils
from src.utils import save_load_utils
from src.utils import plotting_functions

# Import specific functions/classes
from src.utils.config import Config
from src.models import get_model
from src.optimizers import get_opt
from src.metrics import get_metrics

__all__ = ["Config", "get_model", "get_opt", "get_metrics",
           "utils", "save_load_utils",  "plotting_functions",]

# Now can do 
#   "from src import get_model" 
# instead of 
#   "from src.models import get_model"