Skip to content

Commit aeca6f9

Browse files
committed
Add compute advisory todo V2 to improver registry
Signed-off-by: Tushar Goel <tushar.goel.dav@gmail.com>
1 parent 2f4415d commit aeca6f9

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

vulnerabilities/improvers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
enhance_with_metasploit as enhance_with_metasploit_v2,
3030
)
3131
from vulnerabilities.pipelines.v2_improvers import flag_ghost_packages as flag_ghost_packages_v2
32+
from vulnerabilities.pipelines.v2_improvers import compute_advisory_todo as compute_advisory_todo_v2
3233
from vulnerabilities.utils import create_registry
3334

3435
IMPROVERS_REGISTRY = create_registry(
@@ -65,6 +66,7 @@
6566
enhance_with_metasploit_v2.MetasploitImproverPipeline,
6667
compute_package_risk_v2.ComputePackageRiskPipeline,
6768
compute_version_rank_v2.ComputeVersionRankPipeline,
69+
compute_advisory_todo_v2.ComputeToDo,
6870
compute_advisory_todo.ComputeToDo,
6971
]
7072
)

vulnerabilities/pipelines/v2_improvers/compute_advisory_todo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
class ComputeToDo(VulnerableCodePipeline):
2626
"""Compute ToDos for Advisory."""
2727

28-
pipeline_id = "compute_advisory_todo"
28+
pipeline_id = "compute_advisory_todo_v2"
2929

3030
@classmethod
3131
def steps(cls):

0 commit comments

Comments
 (0)