update dataset name to our organization
Browse files- data_loader.py +1 -1
data_loader.py
CHANGED
|
@@ -8,7 +8,7 @@ REMOVED_COMMITS = ['9cc896202dc38d962c01aa2637dbc5bbc3e3dd9b']
|
|
| 8 |
|
| 9 |
|
| 10 |
def load_data():
|
| 11 |
-
df = load_dataset("
|
| 12 |
split="train",
|
| 13 |
token=os.environ.get('HF_REWRITING_TOKEN'),
|
| 14 |
cache_dir=CACHE_DIR).to_pandas()
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
def load_data():
|
| 11 |
+
df = load_dataset("JetBrains-Research/commit-rewriting-samples",
|
| 12 |
split="train",
|
| 13 |
token=os.environ.get('HF_REWRITING_TOKEN'),
|
| 14 |
cache_dir=CACHE_DIR).to_pandas()
|