From fd0a2e8e3d40addae04880f5c8a1015e72a35ab9 Mon Sep 17 00:00:00 2001 From: Tianhao-Gu Date: Mon, 18 Dec 2023 12:46:43 -0600 Subject: [PATCH] make HEATMAP_TO_ATTRIBS_MAPPING private --- src/common/product_models/heatmap_common_models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/common/product_models/heatmap_common_models.py b/src/common/product_models/heatmap_common_models.py index a3bd4a4eb..041b82497 100644 --- a/src/common/product_models/heatmap_common_models.py +++ b/src/common/product_models/heatmap_common_models.py @@ -54,7 +54,7 @@ class ColumnType(str, Enum): # Maps a heatmap column type to a genome attributes column type -HEATMAP_TO_ATTRIBS_MAPPING = { +_HEATMAP_TO_ATTRIBS_MAPPING = { ColumnType.FLOAT: AttribsColumnType.FLOAT, ColumnType.INT: AttribsColumnType.INT, ColumnType.COUNT: AttribsColumnType.INT, @@ -70,10 +70,10 @@ def trans_column_type_heatmap_to_attribs(col_type: ColumnType) -> AttribsColumnT col_type: the heatmap column type """ - if col_type not in HEATMAP_TO_ATTRIBS_MAPPING: + if col_type not in _HEATMAP_TO_ATTRIBS_MAPPING: raise ValueError(f'column type {col_type} is not supported by the heatmap') - return HEATMAP_TO_ATTRIBS_MAPPING[col_type] + return _HEATMAP_TO_ATTRIBS_MAPPING[col_type] class ColumnInformation(BaseModel):