try:
    import nsml

except:
    from tensorfn import nsml_wrapper as nsml

from tensorfn.util import (
    read_config,
    preset_argparser,
    load_config,
    load_arg_config,
    add_distributed_args,
    load_wandb,
    ensure_tuple,
    get_logger,
    create_small_table,
)
from tensorfn.checker import Checker
from tensorfn.trainer import Trainer

try:
    import nsml

except:
    from tensorfn import nsml_wrapper as nsml

from tensorfn.util import (
    read_config,
    preset_argparser,
    load_config,
    load_arg_config,
    add_distributed_args,
    load_wandb,
    ensure_tuple,
    get_logger,
    create_small_table,
)
from tensorfn.checker import Checker
from tensorfn.trainer import Trainer
