# Path: tensorflow_transform/common.py
# ANALYZER_COLLECTION = 'tft_analyzer_use'
# MAPPER_COLLECTION = 'tft_mapper_use'
# ANNOTATION_PREFIX_URL = 'type.googleapis.com'
#   IS_ANNOTATIONS_PB_AVAILABLE = True
#   IS_ANNOTATIONS_PB_AVAILABLE = False
# def logging_context() -> Generator[None, None, None]:
# def log_api_use(
#     collection_name: str) -> Callable[[Callable[..., Any]], Callable[..., Any]]:
#   def decorator(fn):
#     def wrapped_fn(*args, **kwargs):


import copyreg
import tensorflow as tf
from tensorflow_transform import common
from tensorflow_metadata.proto.v0 import schema_pb2
from tensorflow_metadata.proto.v0 import statistics_pb2
  from tensorflow_transform import annotations_pb2  # pylint: disable=g-import-not-at-top
# Copyright 2019 Google Inc. All Rights Reserved.
#
# 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.
"""Functions to fix pickling of certain objects (see b/121323638)."""


if common.IS_ANNOTATIONS_PB_AVAILABLE:

_ANNOTATION_CLASSES = [
    annotations_pb2.VocabularyMetadata, annotations_pb2.BucketBoundaries

# Path: tensorflow_transform/common.py
# ANALYZER_COLLECTION = 'tft_analyzer_use'
# MAPPER_COLLECTION = 'tft_mapper_use'
# ANNOTATION_PREFIX_URL = 'type.googleapis.com'
#   IS_ANNOTATIONS_PB_AVAILABLE = True
#   IS_ANNOTATIONS_PB_AVAILABLE = False
# def logging_context() -> Generator[None, None, None]:
# def log_api_use(
#     collection_name: str) -> Callable[[Callable[..., Any]], Callable[..., Any]]:
#   def decorator(fn):
#     def wrapped_fn(*args, **kwargs):


# Path: tensorflow_transform/pickle_helper.py
import copyreg
import tensorflow as tf
from tensorflow_transform import common
from tensorflow_metadata.proto.v0 import schema_pb2
from tensorflow_metadata.proto.v0 import statistics_pb2
  from tensorflow_transform import annotations_pb2  # pylint: disable=g-import-not-at-top

# Copyright 2019 Google Inc. All Rights Reserved.
#
# 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.
"""Functions to fix pickling of certain objects (see b/121323638)."""


if common.IS_ANNOTATIONS_PB_AVAILABLE:

_ANNOTATION_CLASSES = [
    annotations_pb2.VocabularyMetadata, annotations_pb2.BucketBoundaries

] if common.IS_ANNOTATIONS_PB_AVAILABLE else []