Skip to content

Commit

Permalink
Update oasst-data docs to reflect code changes (#2660)
Browse files Browse the repository at this point in the history
Close #2639
  • Loading branch information
olliestanley committed Apr 17, 2023
1 parent fe61bff commit 95a6d53
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions oasst-data/README.md
Expand Up @@ -19,12 +19,12 @@ Code example:

```python
# parsing OA data files with oasst_data helpers
from oasst_data import load_trees, visit_messages_depth_first, ExportMessageNode
from oasst_data import read_message_trees, visit_messages_depth_first, ExportMessageNode

messages: list[ExportMessageNode] = []

input_file_path = "data_file.jsonl.gz"
for tree in load_trees(input_file_path):
for tree in read_message_trees(input_file_path):
if tree.prompt.lang not in ["en","es"]: # filtering by language tag (optional)
continue

Expand Down

0 comments on commit 95a6d53

Please sign in to comment.