Merge pull request #5 from FujitsuResearch/feature/pypi

v1.0.2: Fix ImportError when matplotlib is not installed

Author: FKKimura

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Change log
 
+## [v1.0.2] 2026-03-31
+
+### Bug Fix
+
+- Fixed `ImportError` when running `onecomp` CLI without matplotlib installed; `AutoBitQuantizer._visualize()` now catches the import error and logs a warning instead of crashing
+
 ## [v1.0.1] 2026-03-31
 
 ### Packaging

onecomp/__version__.py

@@ -6,4 +6,4 @@
 
 """
 
-__version__ = "1.0.1"
+__version__ = "1.0.2"

onecomp/quantizer/autobit/_autobit.py

@@ -402,7 +402,14 @@ def _assign_layer(self, name, module, child_q):
         child_q.module_to_name[module] = name
 
     def _visualize(self, assignments):
-        from .visualize import visualize_bit_assignment  # lazy: heavy matplotlib dep
+        try:
+            from .visualize import visualize_bit_assignment
+        except ImportError:
+            self.logger.warning(
+                "matplotlib is not installed; skipping visualization. "
+                "Install with: pip install onecomp[visualize]"
+            )
+            return
 
         layer_names = [name for name, _, _ in assignments]
         layer_bits = [