From f21d61ae5dc22ca0683c05071c304b63e2ae60b4 Mon Sep 17 00:00:00 2001 From: jeromeku Date: Fri, 28 Mar 2025 16:54:38 -0700 Subject: [PATCH] move hf hub utils to unsloth/utils --- pyproject.toml | 4 ++++ tests/test_model_registry.py | 3 +-- unsloth/utils/__init__.py | 0 {tests => unsloth}/utils/hf_hub.py | 0 4 files changed, 5 insertions(+), 2 deletions(-) create mode 100644 unsloth/utils/__init__.py rename {tests => unsloth}/utils/hf_hub.py (100%) diff --git a/pyproject.toml b/pyproject.toml index 7f24aabbf2..808a956c89 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,6 +32,10 @@ include-package-data = false exclude = ["images*", "tests*"] [project.optional-dependencies] +dev = [ + "pytest", +] + triton = [ "triton-windows ; platform_system == 'Windows'", ] diff --git a/tests/test_model_registry.py b/tests/test_model_registry.py index c3eb4b0c8d..183edc92d5 100644 --- a/tests/test_model_registry.py +++ b/tests/test_model_registry.py @@ -11,8 +11,7 @@ from unsloth.model_registry import ( get_qwen_models, get_qwen_vl_models, ) - -from .utils.hf_hub import get_model_info +from unsloth.utils.hf_hub import get_model_info MODEL_NAMES = [ "llama", diff --git a/unsloth/utils/__init__.py b/unsloth/utils/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/tests/utils/hf_hub.py b/unsloth/utils/hf_hub.py similarity index 100% rename from tests/utils/hf_hub.py rename to unsloth/utils/hf_hub.py