Skip to content

Commit

Permalink
Refactor HeatmapFiles class to include make_heatmaps property in desi…
Browse files Browse the repository at this point in the history
…gn.py
  • Loading branch information
alsmith151 committed Apr 26, 2024
1 parent c00b1bc commit 2e2f122
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion seqnado/design.py
Original file line number Diff line number Diff line change
Expand Up @@ -941,6 +941,7 @@ def files(self) -> List[str]:

class HeatmapFiles(BaseModel):
assay: Literal["ChIP", "ATAC", "RNA", "SNP"]
make_heatmaps: bool = False

Check warning on line 944 in seqnado/design.py

View check run for this annotation

Codecov / codecov/patch

seqnado/design.py#L944

Added line #L944 was not covered by tests

@property
def heatmap_files(self) -> List[str]:
Expand All @@ -952,7 +953,10 @@ def heatmap_files(self) -> List[str]:
@computed_field
@property
def files(self) -> List[str]:
return self.heatmap_files
if self.make_heatmaps:
return self.heatmap_files

Check warning on line 957 in seqnado/design.py

View check run for this annotation

Codecov / codecov/patch

seqnado/design.py#L956-L957

Added lines #L956 - L957 were not covered by tests
else:
return []

Check warning on line 959 in seqnado/design.py

View check run for this annotation

Codecov / codecov/patch

seqnado/design.py#L959

Added line #L959 was not covered by tests


class HubFiles(BaseModel):
Expand Down

0 comments on commit 2e2f122

Please sign in to comment.