diff --git a/gcca/gcca.py b/gcca/gcca.py index 46b611b..e1f712f 100644 --- a/gcca/gcca.py +++ b/gcca/gcca.py @@ -151,8 +151,8 @@ def transform(self, *x_list): return z_list def fit_transform(self, *x_list): - self.fit(x_list) - self.transform(x_list) + self.fit(*x_list) + return self.transform(*x_list) @staticmethod def normalize(mat): @@ -272,4 +272,4 @@ def main(): gcca.calc_correlations() if __name__=="__main__": - main() \ No newline at end of file + main()