From 44e1f74dba1bdf6f79828321ea29529fd22aad19 Mon Sep 17 00:00:00 2001 From: Achal Dave Date: Fri, 8 Dec 2023 20:20:44 -0500 Subject: [PATCH] Import from open_lm, no relative imports (#143) --- open_lm/file_utils.py | 2 +- open_lm/main.py | 14 +++++++------- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/open_lm/file_utils.py b/open_lm/file_utils.py index c4afa9fe..275eea7d 100644 --- a/open_lm/file_utils.py +++ b/open_lm/file_utils.py @@ -16,7 +16,7 @@ from typing import List, Optional from tqdm import tqdm -from .distributed import is_master +from open_lm.distributed import is_master def remote_sync_s3(local_dir, remote_dir): diff --git a/open_lm/main.py b/open_lm/main.py index 0746d181..898c9c53 100644 --- a/open_lm/main.py +++ b/open_lm/main.py @@ -45,13 +45,13 @@ from open_lm.model import create_model from open_lm.utils.transformers.hf_wrapper import create_wrapped_hf_model -from .data import get_data, get_wds_dataset -from .distributed import is_master, init_distributed_device, broadcast_object -from .logger import setup_logging -from .params import parse_args -from .scheduler import cosine_lr -from .train import train_one_epoch, evaluate -from .file_utils import ( +from open_lm.data import get_data, get_wds_dataset +from open_lm.distributed import is_master, init_distributed_device, broadcast_object +from open_lm.logger import setup_logging +from open_lm.params import parse_args +from open_lm.scheduler import cosine_lr +from open_lm.train import train_one_epoch, evaluate +from open_lm.file_utils import ( pt_load, check_exists, start_sync_process,