def set_file_paths()

in data.py [0:0]

• 19 lines of code
• 9 McCabe index (conditional complexity)

    def set_file_paths(self, args):
        """
        Set correct file paths
        :param args:
        :return:
        """
        if not args.pca_file.endswith(".pkl"):
            args.pca_file = args.data_loc + args.data_name + "_train_pca.pkl"
        if not args.model_response_pre.endswith(".pkl"):
            args.model_response_pre = args.data_loc + "{}_{}_store.pkl".format(
                args.data_name, args.mode
            )
        # if args.exp_data_folder == "na":
        #     args.exp_data_folder = os.path.join(args.data_loc, "{}_data".format(args.mode), args.id)
        #     if not os.path.exists(args.exp_data_folder):
        #         os.mkdir(args.exp_data_folder)
        finetuned = ""
        if args.load_fine_tuned:
            finetuned = "_finetuned_" + args.trained_bert_suffix
        if not args.emb_file.endswith(".pkl"):
            args.emb_file = (
                args.emb_file + args.data_name + finetuned + "_{}.pkl".format(args.mode)
            )
        # if not args.data_loc.endswith('.pkl'):
        #     args.data_loc = args.data_loc + args.data_name + \
        #                     '_{}.pkl'.format(args.mode)
        if not args.tok_file.endswith(".pkl"):
            args.tok_file = (
                args.data_loc + args.data_name + "_tokens_{}.pkl".format(args.mode)
            )
        return args