mils2004 commited on
Commit
adf640c
·
1 Parent(s): e9ec293

i dont remember adding security but there are secutity issues

Browse files
Files changed (1) hide show
  1. rag_pipeline/data/load_data.py +3 -3
rag_pipeline/data/load_data.py CHANGED
@@ -15,7 +15,7 @@ def clean_text(text):
15
 
16
 
17
  def load_arxiv_subset(n: int = 10000):
18
- dataset = load_dataset("ccdv/arxiv-summarization")
19
  return dataset["train"].select(range(n))
20
 
21
 
@@ -39,7 +39,7 @@ def process_dataset(data) -> list[dict]:
39
  if len(chunk) < 50:
40
  continue
41
  all_chunks.append({
42
- "chunk_id": hashlib.md5(f"{i}_abstract_{chunk_idx}".encode()).hexdigest(),
43
  "paper_id": i,
44
  "section_title": "ABSTRACT",
45
  "chunk_index": chunk_idx,
@@ -54,7 +54,7 @@ def process_dataset(data) -> list[dict]:
54
  if len(chunk) < 50:
55
  continue
56
  all_chunks.append({
57
- "chunk_id": hashlib.md5(f"{i}_article_{chunk_idx}".encode()).hexdigest(),
58
  "paper_id": i,
59
  "section_title": "ARTICLE",
60
  "chunk_index": chunk_idx,
 
15
 
16
 
17
  def load_arxiv_subset(n: int = 10000):
18
+ dataset = load_dataset("ccdv/arxiv-summarization",revision="main")
19
  return dataset["train"].select(range(n))
20
 
21
 
 
39
  if len(chunk) < 50:
40
  continue
41
  all_chunks.append({
42
+ "chunk_id": hashlib.md5(f"{i}_abstract_{chunk_idx}".encode(),usedforsecurity=False).hexdigest(),
43
  "paper_id": i,
44
  "section_title": "ABSTRACT",
45
  "chunk_index": chunk_idx,
 
54
  if len(chunk) < 50:
55
  continue
56
  all_chunks.append({
57
+ "chunk_id": hashlib.md5(f"{i}_article_{chunk_idx}".encode(),usedforsecurity=False).hexdigest(),
58
  "paper_id": i,
59
  "section_title": "ARTICLE",
60
  "chunk_index": chunk_idx,