baecce5a6fbe4576ca8348a93dc56f07e7ba84f6,niftynet/contrib/csv_reader/csv_reader.py,CSVReader,_parse_csv,#CSVReader#Any#Any#,72

Before Change

                "The first column of the csv should be called "subject_id" and there should be at least 2 columns"
            )
        df.index = df["subject_id"]
        assert set(df.index) == set(self.subject_ids)
        df = df.loc[self.subject_ids]
        if to_ohe and len(df.columns[1:])==1:
            _dims = len(list(df[df.columns[1]].unique()))
            _indexable_output = self.to_ohe(df["label"].values, _dims)

After Change

            _dims = len(list(df[1].unique()))
            _indexable_output = self.to_ohe(df[1].values, _dims)
            return df, _indexable_output, _dims
        elif not to_ohe and len(df.columns==1):
            _dims = 1
            _indexable_output = self.to_categorical(df[1].values, df[1].unique())
            return df, _indexable_output, _dims
        elif not to_ohe:
            _dims = len(df.columns)
            _indexable_output = list(df.values)
            return df, _indexable_output, _dims
        else:
            tf.logging.fatal("Unrecognised input format for {}".format(path_to_csv))

    @staticmethod

Instances