karma/crab
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab.git synced 2025-07-05 13:38:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

moved enums to top of file

Browse Source
This commit is contained in:
Karma Riuk
2025-05-16 19:11:26 +02:00
parent 25161c4d46
commit b84ea797ff
1 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
dataset.py
View File

@ -3,6 +3,18 @@ from enum import Enum
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
import json, argparse, os, uuid import json, argparse, os, uuid
class OutputType(Enum):
FULL = "full"
CODE_REFINEMENT = "code_refinement"
COMMENT_GEN = "comment_gen"
class ArchiveState(Enum):
BASE = "base"
MERGED = "merged"
# fmt: off # fmt: off
@dataclass @dataclass
class FileData: class FileData:
@ -25,10 +37,6 @@ class Selection:
diff_after_address_change: Optional[bool] diff_after_address_change: Optional[bool]
is_code_related: bool is_code_related: bool
class ArchiveState(Enum):
BASE = "base"
MERGED = "merged"
@dataclass @dataclass
class Metadata: class Metadata:
id: str id: str
@ -88,11 +96,6 @@ class CodeRefinementEntry:
comments=entry.comments, comments=entry.comments,
) )
class OutputType(Enum):
FULL = "full"
CODE_REFINEMENT = "code_refinement"
COMMENT_GEN = "comment_gen"
# fmt: on # fmt: on
@dataclass @dataclass
class Dataset: class Dataset: