derek-thomas HF staff commited on
Commit
83f6dc4
1 Parent(s): c422202

Updating to capture nsfw information

Browse files
utilities/praw_downloader.py CHANGED
@@ -34,6 +34,7 @@ def extract_submission_data(submission: praw.models.Submission) -> Dict[str, Any
34
  "title": submission.title,
35
  "score": submission.ups,
36
  "permalink": submission.permalink,
 
37
  }
38
 
39
 
 
34
  "title": submission.title,
35
  "score": submission.ups,
36
  "permalink": submission.permalink,
37
+ "nsfw": submission.over_18,
38
  }
39
 
40
 
utilities/user_defined_functions.py CHANGED
@@ -38,6 +38,7 @@ dummy_data = {
38
  "permalink": ["/r/sampleSubreddit/comments/sampleID/sample_post_title_how_to_use_hugging_face/"],
39
  "updated": [False],
40
  "new": [False],
 
41
  }
42
 
43
 
 
38
  "permalink": ["/r/sampleSubreddit/comments/sampleID/sample_post_title_how_to_use_hugging_face/"],
39
  "updated": [False],
40
  "new": [False],
41
+ "nsfw": [False]
42
  }
43
 
44