Better typing

jonbrenas · jonbrenas · commit 4d0fac4447b1 · 2024-12-13T19:07:54.000Z
diff --git a/malariagen_data/anoph/dipclust.py b/malariagen_data/anoph/dipclust.py
@@ -604,7 +604,7 @@ def plot_diplotype_clustering_advanced(
         region: base_params.regions,
         heterozygosity: bool = True,
         heterozygosity_colorscale: plotly_params.color_continuous_scale = "Greys",
-        snp_transcript: dipclust_params.snp_transcript = None,
+        snp_transcript: Optional[dipclust_params.snp_transcript] = None,
         snp_colorscale: plotly_params.color_continuous_scale = "Greys",
         snp_filter_min_maf: float = 0.05,
         snp_query: Optional[base_params.snp_query] = AA_CHANGE_QUERY,
diff --git a/malariagen_data/anoph/dipclust_params.py b/malariagen_data/anoph/dipclust_params.py
@@ -12,6 +12,6 @@
 distance_metric_default: distance_metric = "cityblock"
 
 snp_transcript: TypeAlias = Annotated[
-    Union[None, transcript, Sequence[transcript]],
+    Union[transcript, Sequence[transcript]],
     "A transcript or a list of transcripts",
 ]

Original file line number	Diff line number	Diff line change
`@@ -12,6 +12,6 @@`
`12`	`12`	`distance_metric_default: distance_metric = "cityblock"`
`13`	`13`
`14`	`14`	`snp_transcript: TypeAlias = Annotated[`
`15`		`- Union[None, transcript, Sequence[transcript]],`
	`15`	`+ Union[transcript, Sequence[transcript]],`
`16`	`16`	`"A transcript or a list of transcripts",`
`17`	`17`	`]`