expose helper function

neuralmagic · Apr 8, 2024 · 5ef7fbf · 5ef7fbf
1 parent 62527af
commit 5ef7fbf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/sparsetensors/utils/__init__.py b/src/sparsetensors/utils/__init__.py
@@ -13,4 +13,5 @@
 # limitations under the License.
 # flake8: noqa
 
+from .helpers import *
 from .safetensors_load import *
diff --git a/src/sparsetensors/utils/helpers.py b/src/sparsetensors/utils/helpers.py
@@ -15,9 +15,9 @@
 
 from typing import Optional
 
+from sparsetensors.base import SPARSITY_CONFIG_NAME
 from sparsetensors.compressors import ModelCompressor
 from sparsetensors.config import CompressionConfig
-from sparsetensors.utils.helpers import SPARSITY_CONFIG_NAME
 from transformers import AutoConfig