Skip to content

Commit

Permalink
Merge pull request #22 from Framartin/fix-doc
Browse files Browse the repository at this point in the history
Fix docstrings in mimir/config.py
  • Loading branch information
iamgroot42 committed Apr 26, 2024
2 parents f553c3d + 625b506 commit 003b2df
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions mimir/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ class ExperimentConfig(Serializable):
max_substrs: Optional[int] = 20
"""If full_doc, determines the maximum number of sample substrs to evaluate on"""
dump_cache: Optional[bool] = False
"Dump data to cache? Exits program after dumping"
"""Dump data to cache? Exits program after dumping"""
load_from_cache: Optional[bool] = False
"""Load data from cache?"""
load_from_hf: Optional[bool] = True
Expand Down Expand Up @@ -187,9 +187,9 @@ class ExperimentConfig(Serializable):
pre_perturb_span_length: Optional[int] = 5
"""Span length for pre-perturbation"""
tok_by_tok: Optional[bool] = False
"""FPRs at which to compute TPR"""
fpr_list: Optional[List[float]] = field(default_factory=lambda: [0.001, 0.01])
"""Process data token-wise?"""
fpr_list: Optional[List[float]] = field(default_factory=lambda: [0.001, 0.01])
"""FPRs at which to compute TPR"""
random_seed: Optional[int] = 0
"""Random seed"""
ref_config: Optional[ReferenceConfig] = None
Expand Down

0 comments on commit 003b2df

Please sign in to comment.