mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
formatting
This commit is contained in:
parent
98988c8481
commit
20069b2a7d
37 changed files with 504 additions and 666 deletions
|
|
@ -4,34 +4,37 @@ from importlib import resources
|
|||
from pathlib import Path
|
||||
from typing import Union
|
||||
|
||||
|
||||
def get_data_file_path(filename: str) -> Path:
|
||||
"""Get the path to a data file in the package.
|
||||
|
||||
|
||||
Args:
|
||||
filename: Name of the file in the data directory
|
||||
|
||||
|
||||
Returns:
|
||||
Path object pointing to the data file
|
||||
|
||||
|
||||
Example:
|
||||
>>> path = get_data_file_path("pg19362.txt")
|
||||
>>> with open(path) as f:
|
||||
... content = f.read()
|
||||
"""
|
||||
return resources.files('reasoning_gym.data').joinpath(filename)
|
||||
return resources.files("reasoning_gym.data").joinpath(filename)
|
||||
|
||||
|
||||
def read_data_file(filename: str) -> str:
|
||||
"""Read the contents of a data file in the package.
|
||||
|
||||
|
||||
Args:
|
||||
filename: Name of the file in the data directory
|
||||
|
||||
|
||||
Returns:
|
||||
String contents of the file
|
||||
|
||||
|
||||
Example:
|
||||
>>> content = read_data_file("pg19362.txt")
|
||||
"""
|
||||
return resources.files('reasoning_gym.data').joinpath(filename).read_text()
|
||||
return resources.files("reasoning_gym.data").joinpath(filename).read_text()
|
||||
|
||||
__all__ = ['get_data_file_path', 'read_data_file']
|
||||
|
||||
__all__ = ["get_data_file_path", "read_data_file"]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue