diff --git a/setup.py b/setup.py index 3d280a2e..999677d1 100644 --- a/setup.py +++ b/setup.py @@ -32,7 +32,7 @@ def _setup_extras() -> Dict: setup( name="compressed-tensors", - version="0.3.2", + version="0.3.3", author="Neuralmagic, Inc.", author_email="support@neuralmagic.com", license="Apache 2.0", diff --git a/src/compressed_tensors/compressors/sparse_bitmask.py b/src/compressed_tensors/compressors/sparse_bitmask.py index 10b398c1..c0cc1888 100644 --- a/src/compressed_tensors/compressors/sparse_bitmask.py +++ b/src/compressed_tensors/compressors/sparse_bitmask.py @@ -67,7 +67,7 @@ def compress(self, model_state: Dict[str, Tensor]) -> Dict[str, Tensor]: f"found an existing entry for {key}. The existing entry will " "be replaced." ) - compressed_dict |= bitmask_dict + compressed_dict.update(bitmask_dict) return compressed_dict diff --git a/tests/compressed_tensors/quantization/observers/quantization/__init__.py b/tests/test_quantization/__init__.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/__init__.py rename to tests/test_quantization/__init__.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/__init__.py b/tests/test_quantization/lifecycle/__init__.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/__init__.py rename to tests/test_quantization/lifecycle/__init__.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/conftest.py b/tests/test_quantization/lifecycle/conftest.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/conftest.py rename to tests/test_quantization/lifecycle/conftest.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_apply.py b/tests/test_quantization/lifecycle/test_apply.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_apply.py rename to tests/test_quantization/lifecycle/test_apply.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_dynamic_lifecycle.py b/tests/test_quantization/lifecycle/test_dynamic_lifecycle.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_dynamic_lifecycle.py rename to tests/test_quantization/lifecycle/test_dynamic_lifecycle.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_forward.py b/tests/test_quantization/lifecycle/test_forward.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_forward.py rename to tests/test_quantization/lifecycle/test_forward.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_frozen.py b/tests/test_quantization/lifecycle/test_frozen.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_frozen.py rename to tests/test_quantization/lifecycle/test_frozen.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_initialize.py b/tests/test_quantization/lifecycle/test_initialize.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_initialize.py rename to tests/test_quantization/lifecycle/test_initialize.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_lifecycle.py b/tests/test_quantization/lifecycle/test_lifecycle.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/lifecycle/test_lifecycle.py rename to tests/test_quantization/lifecycle/test_lifecycle.py diff --git a/tests/compressed_tensors/quantization/observers/test_min_max.py b/tests/test_quantization/test_observers/test_min_max.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/test_min_max.py rename to tests/test_quantization/test_observers/test_min_max.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/test_quant_args.py b/tests/test_quantization/test_quant_args.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/test_quant_args.py rename to tests/test_quantization/test_quant_args.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/test_quant_config.py b/tests/test_quantization/test_quant_config.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/test_quant_config.py rename to tests/test_quantization/test_quant_config.py diff --git a/tests/compressed_tensors/quantization/observers/quantization/test_quant_scheme.py b/tests/test_quantization/test_quant_scheme.py similarity index 100% rename from tests/compressed_tensors/quantization/observers/quantization/test_quant_scheme.py rename to tests/test_quantization/test_quant_scheme.py