


'''simple docstring'''





import argparse

import torch
from torch import nn

from transformers import PLBartConfig, PLBartForConditionalGeneration, PLBartForSequenceClassification


def 					a__    (				UpperCamelCase_	:			Any	):
   UpperCAmelCase__  :Optional[Any]            =			[
       "encoder.version",
       "decoder.version",
       "model.encoder.version",
       "model.decoder.version",
       "_float_tensor",
       "decoder.output_projection.weight",
   ]
   for k in ignore_keys:
      state_dict.pop(UpperCamelCase__,      UpperCamelCase__	)


def 					a__    (				UpperCamelCase_	:			Dict	):
   UpperCAmelCase__  :Tuple            =			emb.weight.shape
   UpperCAmelCase__  :Optional[Any]            =			nn.Linear(UpperCamelCase__,      UpperCamelCase__,      bias=UpperCamelCase__	)
   UpperCAmelCase__  :int            =			emb.weight.data
   return lin_layer


def 					a__    (				UpperCamelCase_	:			List[Any],      UpperCamelCase_	:			List[str]="uclanlp/plbart-base",      UpperCamelCase_	:			List[Any]=False,      UpperCamelCase_	:			Dict=False	):
   UpperCAmelCase__  :str            =			torch.load(UpperCamelCase__,      map_location='''cpu'''	)["model"]
   remove_ignore_keys_(UpperCamelCase__	)
   UpperCAmelCase__  :Dict            =			state_dict["encoder.embed_tokens.weight"].shape[0]

   UpperCAmelCase__  :Dict            =			PLBartConfig.from_pretrained(UpperCamelCase__,      vocab_size=UpperCamelCase__	)

   UpperCAmelCase__  :List[Any]            =			state_dict["decoder.embed_tokens.weight"]
   if not classification:
      UpperCAmelCase__  :Any            =			PLBartForConditionalGeneration(UpperCamelCase__	)
      model.model.load_state_dict(UpperCamelCase__	)
      if finetuned:
         UpperCAmelCase__  :List[str]            =			make_linear_from_emb(model.model.shared	)

   else:
      UpperCAmelCase__  :Dict            =			{}
      for key, value in state_dict.copy().items():
         if key.startswith('''classification_heads.sentence_classification_head'''	):
            UpperCAmelCase__  :Tuple            =			value
            state_dict.pop(UpperCamelCase__	)
      UpperCAmelCase__  :List[Any]            =			PLBartForSequenceClassification(UpperCamelCase__	)
      model.model.load_state_dict(UpperCamelCase__	)
      model.classification_head.load_state_dict(UpperCamelCase__	)

   return model


if __name__ == "__main__":
    __lowerCamelCase       				=	argparse.ArgumentParser()
    # Required parameters
    parser.add_argument('''fairseq_path''', type=str, help='''model.pt on local filesystem.''')
    parser.add_argument('''pytorch_dump_folder_path''', default=None, type=str, help='''Path to the output PyTorch model.''')
    parser.add_argument(
        '''--hf_config''',
        default='''uclanlp/plbart-base''',
        type=str,
        help='''Which huggingface architecture to use: plbart-base''',
    )
    parser.add_argument('''--finetuned''', action='''store_true''', help='''whether the model is a fine-tuned checkpoint''')
    parser.add_argument(
        '''--classification''', action='''store_true''', help='''whether the model is a classification checkpoint'''
    )
    __lowerCamelCase       				=	parser.parse_args()
    __lowerCamelCase       				=	convert_fairseq_plbart_checkpoint_from_disk(
        args.fairseq_path,
        hf_config_path=args.hf_config,
        finetuned=args.finetuned,
        classification=args.classification,
    )
    model.save_pretrained(args.pytorch_dump_folder_path)






'''simple docstring'''





__lowerCamelCase       				=	[
    ['''ab-e1''', '''ac-e3''', '''ad-e5''', '''bc-e4''', '''bd-e2''', '''be-e6''', '''bh-e12''', '''cd-e2''', '''ce-e4''',
     '''de-e1''', '''df-e8''', '''dg-e5''', '''dh-e10''', '''ef-e3''', '''eg-e2''', '''fg-e6''', '''gh-e6''', '''hi-e3'''],
    ['''ab-e1''', '''ac-e3''', '''ad-e5''', '''bc-e4''', '''bd-e2''', '''be-e6''', '''cd-e2''', '''de-e1''', '''df-e8''',
     '''ef-e3''', '''eg-e2''', '''fg-e6'''],
    ['''ab-e1''', '''ac-e3''', '''bc-e4''', '''bd-e2''', '''de-e1''', '''df-e8''', '''dg-e5''', '''ef-e3''', '''eg-e2''',
     '''eh-e12''', '''fg-e6''', '''fh-e10''', '''gh-e6'''],
    ['''ab-e1''', '''ac-e3''', '''bc-e4''', '''bd-e2''', '''bh-e12''', '''cd-e2''', '''df-e8''', '''dh-e10'''],
    ['''ab-e1''', '''ac-e3''', '''ad-e5''', '''bc-e4''', '''bd-e2''', '''cd-e2''', '''ce-e4''', '''de-e1''', '''df-e8''',
     '''dg-e5''', '''ef-e3''', '''eg-e2''', '''fg-e6''']
]


def 					a__    (				UpperCamelCase_	:			List[str]	):
   UpperCAmelCase__  :str            =			set()
   for row in edge_array:
      for item in row:
         distinct_edge.add(item[0]	)
   return list(UpperCamelCase_	)


def 					a__    (				UpperCamelCase_	:			Tuple,      UpperCamelCase_	:			List[Any]	):
   UpperCAmelCase__  :int            =			['''0'''] * len(UpperCamelCase_	)
   for i, row in enumerate(UpperCamelCase_	):
      for item in row:
         if distinct_edge in item[0]:
            UpperCAmelCase__  :List[Any]            =			'''1'''
            break
   return "".join(UpperCamelCase_	)


def 					a__    (				UpperCamelCase_	:			Optional[int]	):
   UpperCAmelCase__  :Any            =			get_distinct_edge(UpperCamelCase_	)
   UpperCAmelCase__  :Dict            =			{}

   for item in distinct_edge:
      UpperCAmelCase__  :List[str]            =			get_bitcode(UpperCamelCase_,      UpperCamelCase_	)
      # print('bit',bit)
      # bt=''.join(bit)
      UpperCAmelCase__  :int            =			bit.count('''1'''	)
      UpperCAmelCase__  :Tuple            =			[s, bit]
   # Store [Distinct edge, WT(Bitcode), Bitcode] in descending order
   UpperCAmelCase__  :str            =			[
       [k, v[0], v[1]]
       for k, v in sorted(frequency_table.items(),      key=lambda UpperCamelCase_	: v[1][0],      reverse=UpperCamelCase_	)
   ]
   return sorted_frequency_table


def 					a__    (				UpperCamelCase_	:			Optional[int]	):
   UpperCAmelCase__  :int            =			{}
   for _, item in enumerate(UpperCamelCase_	):
      nodes.setdefault(item[2],      []	).append(item[0]	)
   return nodes


def 					a__    (				UpperCamelCase_	:			Any	):
   UpperCAmelCase__  :Any            =			{}
   for key, value in nodes.items():
      UpperCAmelCase__  :Optional[int]            =			value
   return cluster


def 					a__    (				UpperCamelCase_	:			Tuple	):
   return [i * 100 / len(UpperCamelCase_	) for i in cluster]


def 					a__    (				):
   print('''\nNodes\n'''	)
   for key, value in nodes.items():
      print(UpperCamelCase_,      UpperCamelCase_	)
   print('''\nSupport\n'''	)
   print(UpperCamelCase_	)
   print('''\n Cluster \n'''	)
   for key, value in sorted(cluster.items(),      reverse=UpperCamelCase_	):
      print(UpperCamelCase_,      UpperCamelCase_	)
   print('''\n Graph\n'''	)
   for key, value in graph.items():
      print(UpperCamelCase_,      UpperCamelCase_	)
   print('''\n Edge List of Frequent subgraphs \n'''	)
   for edge_list in freq_subgraph_edge_list:
      print(UpperCamelCase_	)


def 					a__    (				UpperCamelCase_	:			Optional[int],      UpperCamelCase_	:			Any,      UpperCamelCase_	:			int,      UpperCamelCase_	:			Dict	):
   for i in cluster[ca]:
      UpperCAmelCase__  :int            =			0
      UpperCAmelCase__  :int            =			ca + 1
      while ca < max(cluster.keys()	):
         for j in cluster[ca]:
            if int(UpperCamelCase_,      2	) & int(UpperCamelCase_,      2	) == int(UpperCamelCase_,      2	):
               if tuple(nodes[i]	) in graph:
                  graph[tuple(nodes[i]	)].append(nodes[j]	)
               else:
                  UpperCAmelCase__  :int            =			[nodes[j]]
               count += 1
         if count == 0:
            UpperCAmelCase__  :List[Any]            =			ca + 1
         else:
            break


def 					a__    (				UpperCamelCase_	:			Optional[int],      UpperCamelCase_	:			List[str]	):
   UpperCAmelCase__  :List[Any]            =			cluster[max(cluster.keys()	)]
   UpperCAmelCase__  :Union[str, Any]            =			'''Header'''
   UpperCAmelCase__  :Optional[int]            =			{}
   for i in x:
      if (["Header"],) in graph:
         graph[(["Header"],)].append(x[i]	)
      else:
         UpperCAmelCase__  :Tuple            =			[x[i]]
   for i in x:
      UpperCAmelCase__  :Dict            =			[['''Header''']]
   UpperCAmelCase__  :Any            =			1
   while i < max(UpperCamelCase_	) - 1:
      create_edge(UpperCamelCase_,      UpperCamelCase_,      UpperCamelCase_,      UpperCamelCase_	)
      UpperCAmelCase__  :Union[str, Any]            =			i + 1
   return graph


def 					a__    (				UpperCamelCase_	:			Optional[Any],      UpperCamelCase_	:			int,      UpperCamelCase_	:			Any,      UpperCamelCase_	:			Dict=None	):
   UpperCAmelCase__  :Tuple            =			(path or []) + [start]
   if start == end:
      paths.append(UpperCamelCase_	)
   for node in graph[start]:
      if tuple(UpperCamelCase_	) not in path:
         my_dfs(UpperCamelCase_,      tuple(UpperCamelCase_	),      UpperCamelCase_,      UpperCamelCase_	)


def 					a__    (				UpperCamelCase_	:			Dict,      UpperCamelCase_	:			int,      UpperCamelCase_	:			List[Any]	):
   UpperCAmelCase__  :List[str]            =			int(s / 100 * (len(UpperCamelCase_	) - 1)	)
   for i in cluster[k]:
      my_dfs(UpperCamelCase_,      tuple(cluster[k][i]	),      (['''Header'''],)	)


def 					a__    (				UpperCamelCase_	:			List[Any]	):
   UpperCAmelCase__  :int            =			[]
   for edges in paths:
      UpperCAmelCase__  :List[Any]            =			[]
      for j in range(len(UpperCamelCase_	) - 1	):
         UpperCAmelCase__  :Any            =			list(edges[j]	)
         for e in temp:
            UpperCAmelCase__  :str            =			(e[0], e[1])
            el.append(UpperCamelCase_	)
      freq_sub_el.append(UpperCamelCase_	)
   return freq_sub_el


def 					a__    (				UpperCamelCase_	:			Dict	):
   for i in range(len(UpperCamelCase_	)	):
      for j in range(len(edge_array[i]	)	):
         UpperCAmelCase__  :Dict            =			edge_array[i][j].split('''-'''	)
         UpperCAmelCase__  :Optional[Any]            =			t


if __name__ == "__main__":
    preprocess(edge_array)
    __lowerCamelCase       				=	get_frequency_table(edge_array)
    __lowerCamelCase       				=	get_nodes(frequency_table)
    __lowerCamelCase       				=	get_cluster(nodes)
    __lowerCamelCase       				=	get_support(cluster)
    __lowerCamelCase       				=	construct_graph(cluster, nodes)
    find_freq_subgraph_given_support(60, cluster, graph)
    __lowerCamelCase       				=	[]
    __lowerCamelCase       				=	freq_subgraphs_edge_list(paths)
    print_all()



