# Path: plur/stage_1/cubert_variable_misuse_repair_dataset.py
# _BEAM_METRIC_NAMESPACE = 'cubert_variable_misuse_repair_dataset'
#   ALL_CLASSES = frozenset((
#       'Correct',
#       'Variable misuse',
#   ))
#   _CONFOUNDING_STRING = '###DO_NOT_COPY_ME###'
# class CuBertVariableMisuseRepairDataset(cubert_dataset.CuBertDataset):
#   def __init__(self,
#                stage_1_dir,
#                configuration: plur_dataset.Configuration,
#                *args,
#                use_pointer_output: bool = True,
#                allow_output_copy: bool = True,
#                **kwargs) -> None:
#   def folder_path(self) -> str:
#   def dataset_name(self) -> str:
#   def dataset_description(self) -> str:
#   def data_to_graph_to_output_example(
#       self, data: Mapping[str, Any],
#       max_graph_size: int,
#       split: str) -> Optional[GraphToOutputExample]:
# 
# Path: plur/stage_1/plur_dataset.py
# class Configuration():
# class PlurDataset(abc.ABC):
#   def __init__(self, dataset_name: str, urls: Dict[str, Dict[str, str]],
#                git_url: Dict[str, str], dataset_description: str,
#                stage_1_dir: str,
#                configuration: Configuration = Configuration(),
#                transformation_funcs=(), filter_funcs=(),
#                user_defined_split_range=(), num_shards=1000, seed=0,
#                deduplicate=False):
#   def stage_1_mkdirs(self):
#   def download_dataset(self):
#   def download_dataset_using_git(self):
#   def download_dataset_using_requests(self):
#     def _get_confirm_token(response):
#   def get_random_split(self):
#   def get_all_raw_data_paths(self):
#   def raw_data_paths_to_raw_data_do_fn(self):
#   def raw_data_to_graph_to_output_example(self, raw_data):
#   def exists_graph_to_output_example(self):
#   def convert_and_write_pipeline(self, root):
#   def update_dataset_metadata(self, metadata, split, raw_data_dicts,
#                               graph_to_output_example_dicts):
#   def _count_graph_to_output_example_with_max_num_nodes(
#       self, graph_to_output_example_dicts, split, max_num_nodes):
#   def run_pipeline(self):


from plur.stage_1 import cubert_variable_misuse_repair_dataset
from plur.stage_1 import plur_dataset
# Copyright 2021 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Converts the CuBERT VMR UNC dataset (no pointer, no token copies) to PLUR."""
# TODO(maniatis): Add a test for this dataset.



class CuBertVariableMisuseRepairUnpointedNoCopyDataset(
    cubert_variable_misuse_repair_dataset.CuBertVariableMisuseRepairDataset):
  """As per superclass, but does not allow output copying or input pointers."""

  def __init__(self,
               stage_1_dir,

# Path: plur/stage_1/cubert_variable_misuse_repair_dataset.py
# _BEAM_METRIC_NAMESPACE = 'cubert_variable_misuse_repair_dataset'
#   ALL_CLASSES = frozenset((
#       'Correct',
#       'Variable misuse',
#   ))
#   _CONFOUNDING_STRING = '###DO_NOT_COPY_ME###'
# class CuBertVariableMisuseRepairDataset(cubert_dataset.CuBertDataset):
#   def __init__(self,
#                stage_1_dir,
#                configuration: plur_dataset.Configuration,
#                *args,
#                use_pointer_output: bool = True,
#                allow_output_copy: bool = True,
#                **kwargs) -> None:
#   def folder_path(self) -> str:
#   def dataset_name(self) -> str:
#   def dataset_description(self) -> str:
#   def data_to_graph_to_output_example(
#       self, data: Mapping[str, Any],
#       max_graph_size: int,
#       split: str) -> Optional[GraphToOutputExample]:
# 
# Path: plur/stage_1/plur_dataset.py
# class Configuration():
# class PlurDataset(abc.ABC):
#   def __init__(self, dataset_name: str, urls: Dict[str, Dict[str, str]],
#                git_url: Dict[str, str], dataset_description: str,
#                stage_1_dir: str,
#                configuration: Configuration = Configuration(),
#                transformation_funcs=(), filter_funcs=(),
#                user_defined_split_range=(), num_shards=1000, seed=0,
#                deduplicate=False):
#   def stage_1_mkdirs(self):
#   def download_dataset(self):
#   def download_dataset_using_git(self):
#   def download_dataset_using_requests(self):
#     def _get_confirm_token(response):
#   def get_random_split(self):
#   def get_all_raw_data_paths(self):
#   def raw_data_paths_to_raw_data_do_fn(self):
#   def raw_data_to_graph_to_output_example(self, raw_data):
#   def exists_graph_to_output_example(self):
#   def convert_and_write_pipeline(self, root):
#   def update_dataset_metadata(self, metadata, split, raw_data_dicts,
#                               graph_to_output_example_dicts):
#   def _count_graph_to_output_example_with_max_num_nodes(
#       self, graph_to_output_example_dicts, split, max_num_nodes):
#   def run_pipeline(self):


# Path: plur/stage_1/cubert_variable_misuse_repair_unpointed_nocopy_dataset.py
from plur.stage_1 import cubert_variable_misuse_repair_dataset
from plur.stage_1 import plur_dataset

# Copyright 2021 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Converts the CuBERT VMR UNC dataset (no pointer, no token copies) to PLUR."""
# TODO(maniatis): Add a test for this dataset.



class CuBertVariableMisuseRepairUnpointedNoCopyDataset(
    cubert_variable_misuse_repair_dataset.CuBertVariableMisuseRepairDataset):
  """As per superclass, but does not allow output copying or input pointers."""

  def __init__(self,
               stage_1_dir,

               configuration: plur_dataset.Configuration,