# Copyright 2022 The TensorFlow Authors. 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.
# ==============================================================================
"""
Optionally imports the ModelCardGenerator TFX component.
"""
import logging

try:
  from tfx_addons.model_card_generator.component import \
      ModelCardGenerator  # pylint: disable=unused-import
except ImportError:
  ModelCardGenerator = None
  logging.info(
      """
      The ModelCardGenerator component and its dependencies have been moved to
      the TFX Addons project (https://github.com/tensorflow/tfx-addons).

      If you'd like to use the component,
      you need to install the tfx-addons package:

      ```
      $ pip install tfx-addons[model_card_generator]
      ```
      """
  )
