Skip to content

Commit

Permalink
fix docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
endast committed Nov 21, 2023
1 parent 6203235 commit d020a40
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 3 deletions.
3 changes: 3 additions & 0 deletions fake_vcf/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,9 @@ def version_callback(print_version: bool) -> None:
Args:
print_version (bool): Flag to print the version.
Raises:
Exit: If the print_version flag is set.
"""
if print_version:
console.print(f"[yellow]fake-vcf[/] version: [bold blue]{version}[/]")
Expand Down
9 changes: 6 additions & 3 deletions fake_vcf/vcf_faker.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,10 +12,10 @@ def __init__(
num_rows: int,
num_samples: int,
chromosome: str,
sample_prefix: str = "SAMPLES",
sample_prefix: Optional[str] = "SAMPLES",
random_seed: Optional[int] = None,
phased: bool = True,
large_format: bool = True,
phased: Optional[bool] = True,
large_format: Optional[bool] = True,
):
"""
Initialize VirtualVCF object.
Expand All @@ -28,6 +28,9 @@ def __init__(
random_seed (int, optional): Random seed for reproducibility. Defaults to None.
phased (bool, optional): Phased or unphased genotypes. Defaults to True.
large_format (bool, optional): Use large format VCF. Defaults to True.
Raises:
ValueError: If num_samples or num_rows is less than 1.
"""
self.num_rows = num_rows
self.rows_remaining = num_rows + 1 # One for the header
Expand Down

0 comments on commit d020a40

Please sign in to comment.