File size: 739 Bytes
305e536
aab3281
305e536
a8a595d
 
0c136d8
a8a595d
0c136d8
305e536
5f3a4af
305e536
 
aab3281
5f3a4af
 
 
 
 
 
 
aab3281
 
 
 
 
 
0c136d8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import os
from pathlib import Path

RANDOM_STATE = 42

GRAZIE_API_JWT_TOKEN = os.environ.get("GRAZIE_API_JWT_TOKEN")
GRAZIE_TIMEOUT_SEC = 1.0

HF_TOKEN = os.environ.get('HF_TOKEN')

HF_RAW_DATASET_NAME = "petrtsv-jb/commit-msg-rewriting"
HF_RAW_DATASET_SPLIT = 'train'

HF_FULL_COMMITS_DATASET_NAME = "JetBrains-Research/lca-commit-message-generation"
HF_FULL_COMMITS_DATASET_SUBNAME = "commitchronicle-py-long"
HF_FULL_COMMITS_DATASET_SPLIT = "test"

HF_SYNTHETIC_DATASET_NAME = "petrtsv-jb/synthetic-commit-msg-rewriting"
HF_SYNTHETIC_DATASET_SPLIT = 'train'

CACHE_DIR = Path("cache")
CACHE_DIR.mkdir(exist_ok=True)

OUTPUT_DIR = Path("output")
OUTPUT_DIR.mkdir(exist_ok=True)

SYNTHETIC_DATASET_ARTIFACT = OUTPUT_DIR / "synthetic.csv"