Merge branch 'dev'

aramis-lab · Feb 8, 2021 · fdbe09e · fdbe09e
2 parents 10ef75f + 9305b88
commit fdbe09e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/clinicadl/clinicadl/VERSION b/clinicadl/clinicadl/VERSION
@@ -1 +1 @@
-0.2.0rc2
+0.2.0
diff --git a/clinicadl/clinicadl/tools/tsv/data_split.py b/clinicadl/clinicadl/tools/tsv/data_split.py
@@ -291,7 +291,7 @@ def split_diagnoses(formatted_data_path,
                 # Find similarity of distribution for the age variable
                 if len(set(age)) != 1:
                     age_test = [float(age[idx]) for idx in idx_test]
-                    age_train = [float(age[idx]) for idx in idx_train]
+                    age_train = [float(age[idx]) for idx in idx_train] + sup_train_age
 
                     t_age, p_age = ttest_ind(age_test, age_train)
                 else:
@@ -300,7 +300,7 @@ def split_diagnoses(formatted_data_path,
                 # Find similarity of distribution for the sex variable
                 if len(set(sex)) != 1:
                     sex_test = [sex_dict[sex[idx]] for idx in idx_test]
-                    sex_train = [sex_dict[sex[idx]] for idx in idx_train]
+                    sex_train = [sex_dict[sex[idx]] for idx in idx_train] + sup_train_sex
                     T_sex = chi2(sex_test, sex_train)
                 else:
                     T_sex = 0