Skip to content

Commit

Permalink
Use graphql for LW
Browse files Browse the repository at this point in the history
  • Loading branch information
mruwnik committed Jun 16, 2023
1 parent 3fafc71 commit 1fc2911
Show file tree
Hide file tree
Showing 4 changed files with 143 additions and 453 deletions.
11 changes: 10 additions & 1 deletion align_data/common/alignment_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,9 @@ class AlignmentDataset:
COOLDOWN = 0
"""An optional cool down between processing entries"""

lazy_eval = False
"""Whether to lazy fetch items. This is nice in that it will start processing, but messes up the progress bar."""

# Internal housekeeping variables
_entry_idx = 0
"""Used internally for writing debugging info - each file write will increment it"""
Expand Down Expand Up @@ -142,7 +145,13 @@ def unprocessed_items(self, items=None):
def not_processed(item):
return self.get_item_key(item) not in self._outputted_items

return tqdm(list(filter(not_processed, items or self.items_list)))
filtered = filter(not_processed, items or self.items_list)

# greedily fetch all items if not lazy eval. This makes the progress bar look nice
if not self.lazy_eval:
filtered = list(filtered)

return tqdm(filtered)

def fetch_entries(self):
"""Get all entries to be written to the file."""
Expand Down
15 changes: 12 additions & 3 deletions align_data/greaterwrong/__init__.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,25 @@
from .greaterwrong import GreaterWrong
from .greaterwrong import GreaterWrong, fetch_ea_forum_topics, fetch_LW_tags

GREATERWRONG_REGISTRY = [
GreaterWrong(
name="lesswrong",
base_url='https://www.greaterwrong.com',
base_url='https://www.lesswrong.com',
start_year=2005,
min_karma=1,
af=False,
),
GreaterWrong(
name="alignmentforum",
base_url='https://www.alignmentforum.org',
start_year=2009,
min_karma=1,
af=True,
),
GreaterWrong(
name="eaforum",
base_url='https://ea.greaterwrong.com',
base_url='https://forum.effectivealtruism.org',
start_year=2011,
min_karma=1,
af=False,
)
]
Loading

0 comments on commit 1fc2911

Please sign in to comment.