Add __all__

tanner-andrulis · tanner-andrulis · commit 9a5e07765a8d · 2026-02-01T16:16:09.000-05:00
diff --git a/hwcomponents/__init__.py b/hwcomponents/__init__.py
@@ -8,3 +8,15 @@
     get_leak_power,
     get_model,
 )
+
+__all__ = [
+    "ComponentModel",
+    "action",
+    "get_models",
+    "scaling",
+    "get_area",
+    "get_energy",
+    "get_latency",
+    "get_leak_power",
+    "get_model",
+]
diff --git a/hwcomponents/scaling/__init__.py b/hwcomponents/scaling/__init__.py
@@ -5,3 +5,22 @@
     tech_node_latency,
     tech_node_leak,
 )
+
+__all__ = [
+    # From scalefuncs
+    "linear",
+    "reciprocal",
+    "pow_base",
+    "quadratic",
+    "nlog_base",
+    "nlog2n",
+    "cacti_depth_energy",
+    "cacti_depth_area",
+    "cacti_depth_leak",
+    "noscale",
+    # From techscaling
+    "tech_node_area",
+    "tech_node_energy",
+    "tech_node_latency",
+    "tech_node_leak",
+]
diff --git a/notebooks/model_example/hwcomponents_example/__init__.py b/notebooks/model_example/hwcomponents_example/__init__.py
@@ -1 +1,3 @@
 from .TernaryMAC import TernaryMAC
+
+__all__ = ["TernaryMAC"]

Original file line number	Diff line number	Diff line change
`@@ -1 +1,3 @@`
`1`	`1`	`from .TernaryMAC import TernaryMAC`
	`2`	`+`
	`3`	`+__all__ = ["TernaryMAC"]`