From 5ef7fbfe5844d374a42f47ab4300e3596e607c63 Mon Sep 17 00:00:00 2001
From: dbogunowicz <damian@neuralmagic.com>
Date: Mon, 8 Apr 2024 12:43:44 +0000
Subject: [PATCH] expose helper function

---
 src/sparsetensors/utils/__init__.py | 1 +
 src/sparsetensors/utils/helpers.py  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/sparsetensors/utils/__init__.py b/src/sparsetensors/utils/__init__.py
index 5bc0fec2..e9e78d44 100644
--- 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
index 8d06cde2..c584c2ee 100644
--- 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