restore network editor inports in __init__, remove quant funcions rules from docs

irenab · irenab · commit dc62f9a68cd5 · 2025-06-30T09:48:49.000+03:00
diff --git a/docsrc/source/api/api_docs/modules/network_editor.rst b/docsrc/source/api/api_docs/modules/network_editor.rst
@@ -46,10 +46,6 @@ Actions
 
 |
 
-.. autoclass:: model_compression_toolkit.core.network_editor.ChangeQuantizationParamFunction
-
-|
-
 .. autoclass:: model_compression_toolkit.core.network_editor.ChangeFinalWeightsQuantizationMethod
 
 |
diff --git a/model_compression_toolkit/core/common/network_editors/__init__.py b/model_compression_toolkit/core/common/network_editors/__init__.py
@@ -13,6 +13,14 @@
 # limitations under the License.
 # ==============================================================================
 
+from model_compression_toolkit.core.common.network_editors.actions import (
+    ChangeCandidatesWeightsQuantConfigAttr,
+    ChangeFinalWeightsQuantConfigAttr,
+    ChangeCandidatesActivationQuantConfigAttr,
+    ChangeCandidatesActivationQuantizationMethod,
+    ChangeFinalWeightsQuantizationMethod,
+    ChangeCandidatesWeightsQuantizationMethod,
+    ChangeFinalActivationQuantConfigAttr)
 from model_compression_toolkit.core.common.network_editors.actions import EditRule
 from model_compression_toolkit.core.common.network_editors.node_filters import NodeTypeFilter, NodeNameScopeFilter, \
     NodeNameFilter

Original file line number	Diff line number	Diff line change
`@@ -46,10 +46,6 @@ Actions`
`46`	`46`
`47`	`47`	`\|`
`48`	`48`
`49`		`-.. autoclass:: model_compression_toolkit.core.network_editor.ChangeQuantizationParamFunction`
`50`		`-`
`51`		`-\|`
`52`		`-`
`53`	`49`	`.. autoclass:: model_compression_toolkit.core.network_editor.ChangeFinalWeightsQuantizationMethod`
`54`	`50`
`55`	`51`	`\|`