fix wandb so mypy doesn't complain (#562)
* fix wandb so mypy doesn't complain * fix wandb so mypy doesn't complain * no need for mypy override anymore
This commit is contained in:
@@ -26,7 +26,7 @@ from axolotl.utils.dict import DictDefault
|
||||
from axolotl.utils.distributed import is_main_process
|
||||
from axolotl.utils.models import load_tokenizer
|
||||
from axolotl.utils.tokenization import check_dataset_labels
|
||||
from axolotl.utils.wandb import setup_wandb_env_vars
|
||||
from axolotl.utils.wandb_ import setup_wandb_env_vars
|
||||
|
||||
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
||||
src_dir = os.path.join(project_root, "src")
|
||||
|
||||
Reference in New Issue
Block a user