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
Files changed (1) hide show
  1. src/mosaic/inference/data.py +5 -2
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 8 ("Not Applicable")
303
  """
304
  site_map = get_tissue_site_map()
305
- return site_map.get(site_name, 8)
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):