update imputation_R2

raiss/imputation_R2.py

@@ -29,7 +29,7 @@ def generated_test_data(zscore, N_to_mask=5000, condition=None,  stratifying_vec
         if isinstance(condition, pd.Series)==True:
             masked = np.random.choice(zscore.index[condition], N_to_mask, replace=False)
         else:
-            inter_id = zscore.index.intersection(stratifying_vector.index).drop_duplicates(keep='first', inplace=True)
+            inter_id = zscore.index.intersection(stratifying_vector.index).drop_duplicates(keep='first')
             stratifying_vector = stratifying_vector.loc[inter_id]
             if isinstance(stratifying_vector, pd.Series)==True:
                 masked = []