Add assert that fractional segs are fractional

CPBridge · CPBridge · commit 91d74fd24fa4 · 2025-11-02T22:42:19.000Z
diff --git a/src/idc_annotation_conversion/tcga_til_maps/convert.py b/src/idc_annotation_conversion/tcga_til_maps/convert.py
@@ -221,6 +221,9 @@ def convert_txt_file(
         names=["x", "y", "value", "0"],
     )
 
+    if is_probability:
+        assert df["value"].nunique() > 2
+
     sorted_x_vals = df.sort_values("x").x.unique()
     diff_x = np.diff(sorted_x_vals)
     sorted_y_vals = df.sort_values("y").y.unique()

Original file line number	Diff line number	Diff line change
`@@ -221,6 +221,9 @@ def convert_txt_file(`
`221`	`221`	`names=["x", "y", "value", "0"],`
`222`	`222`	`)`
`223`	`223`
	`224`	`+ if is_probability:`
	`225`	`+ assert df["value"].nunique() > 2`
	`226`	`+`
`224`	`227`	`sorted_x_vals = df.sort_values("x").x.unique()`
`225`	`228`	`diff_x = np.diff(sorted_x_vals)`
`226`	`229`	`sorted_y_vals = df.sort_values("y").y.unique()`