diff --git a/mythril/laser/plugin/plugins/dependency_pruner.py b/mythril/laser/plugin/plugins/dependency_pruner.py index 63171016..a4489a45 100644 --- a/mythril/laser/plugin/plugins/dependency_pruner.py +++ b/mythril/laser/plugin/plugins/dependency_pruner.py @@ -1,5 +1,5 @@ from mythril.laser.ethereum.svm import LaserEVM -from mythril.laser.plugin import LaserPlugin +from mythril.laser.plugin import LaserPlugin, PluginBuilder from mythril.laser.plugin.signals import PluginSkipState from mythril.laser.plugin.plugins.plugin_annotations import ( DependencyAnnotation, @@ -69,6 +69,13 @@ def get_ws_dependency_annotation(state: GlobalState) -> WSDependencyAnnotation: return annotation +class DependencyPrunerBuilder(PluginBuilder): + plugin_name = "dependency-pruner" + + def __call__(self, *args, **kwargs): + return DependencyPruner() + + class DependencyPruner(LaserPlugin): """Dependency Pruner Plugin