Spaces:
Sleeping
Sleeping
copilot-swe-agent[bot]
raylim
commited on
Commit
·
e1efb8d
1
Parent(s):
8141b27
Define DEFAULT_TISSUE_SITE_IDX constant for "Not Applicable" tissue site
Browse files
src/mosaic/inference/data.py
CHANGED
|
@@ -204,6 +204,9 @@ INT_TO_CANCER_TYPE_MAP = {v: k for k, v in CANCER_TYPE_TO_INT_MAP.items()}
|
|
| 204 |
# Tissue site mapping (module-level cache)
|
| 205 |
_TISSUE_SITE_MAP = None
|
| 206 |
|
|
|
|
|
|
|
|
|
|
| 207 |
|
| 208 |
def get_tissue_site_map():
|
| 209 |
"""Load tissue site name → index mapping from CSV.
|
|
@@ -299,10 +302,10 @@ def encode_tissue_site(site_name):
|
|
| 299 |
site_name: Tissue site name from CSV
|
| 300 |
|
| 301 |
Returns:
|
| 302 |
-
int: Tissue site index, defaults to
|
| 303 |
"""
|
| 304 |
site_map = get_tissue_site_map()
|
| 305 |
-
return site_map.get(site_name,
|
| 306 |
|
| 307 |
|
| 308 |
def tissue_site_to_one_hot(site_idx, num_classes=57):
|
|
|
|
| 204 |
# Tissue site mapping (module-level cache)
|
| 205 |
_TISSUE_SITE_MAP = None
|
| 206 |
|
| 207 |
+
# Default tissue site index for "Not Applicable"
|
| 208 |
+
DEFAULT_TISSUE_SITE_IDX = 8
|
| 209 |
+
|
| 210 |
|
| 211 |
def get_tissue_site_map():
|
| 212 |
"""Load tissue site name → index mapping from CSV.
|
|
|
|
| 302 |
site_name: Tissue site name from CSV
|
| 303 |
|
| 304 |
Returns:
|
| 305 |
+
int: Tissue site index, defaults to DEFAULT_TISSUE_SITE_IDX ("Not Applicable")
|
| 306 |
"""
|
| 307 |
site_map = get_tissue_site_map()
|
| 308 |
+
return site_map.get(site_name, DEFAULT_TISSUE_SITE_IDX)
|
| 309 |
|
| 310 |
|
| 311 |
def tissue_site_to_one_hot(site_idx, num_classes=57):
|