fix(benchmarkjava)!: drop attribute vuln_type

Author: nolliv22

codesectools/datasets/BenchmarkJava/dataset.py

@@ -22,17 +22,13 @@ class TestCode(File):
     Inherits from the base `File` class and adds a `vuln_type` attribute
     specific to this dataset.
 
-    Attributes:
-        vuln_type (str): The type of vulnerability present in the file.
-
     """
 
     def __init__(
         self,
         filepath: Path,
         content: str | bytes,
         cwes: list[CWE],
-        vuln_type: str,
         has_vuln: bool,
     ) -> None:
         """Initialize a TestCode instance.
@@ -41,16 +37,13 @@ def __init__(
             filepath: The path to the file.
             content: The content of the file, as a string or bytes.
             cwes: A list of CWEs associated with the file.
-            vuln_type: The type of vulnerability.
             has_vuln: A boolean indicating if the vulnerability is real or a false positive test case.
 
         """
         super().__init__(
             filepath=filepath, content=content, cwes=cwes, has_vuln=has_vuln
         )
 
-        self.vuln_type = vuln_type
-
 
 class BenchmarkJava(PrebuiltFileDataset):
     """Represents the BenchmarkJava dataset.
@@ -151,14 +144,12 @@ def load_dataset(self) -> list[TestCode]:
             filepath = testcode_dir / filename
             content = filepath.read_text()
             cwes = [CWEs.from_id(int(row[3]))]
-            vuln_type = row[1]
             has_vuln = True if row[2] == "true" else False
             files.append(
                 TestCode(
                     filepath.relative_to(self.directory),
                     content,
                     cwes,
-                    vuln_type,
                     has_vuln,
                 )
             )