Skip to content

Commit ce6f947

Browse files
committed
internal inly
PiperOrigin-RevId: 651442932
1 parent 29f0b0c commit ce6f947

File tree

2 files changed

+24
-2
lines changed

2 files changed

+24
-2
lines changed

Diff for: tfx/components/statistics_gen/executor.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
from absl import logging
1919
import tensorflow_data_validation as tfdv
2020
from tensorflow_data_validation.statistics import stats_options as options
21-
from tensorflow_data_validation.utils import dashboard_util
2221
from tfx import types
2322
from tfx.components.statistics_gen import stats_artifact_utils
2423
from tfx.components.util import examples_utils
@@ -28,6 +27,7 @@
2827
from tfx.types import standard_component_specs
2928
from tfx.utils import io_utils
3029
from tfx.utils import json_utils
30+
from tfx.utils import stats_utils
3131

3232

3333
# Default file name for stats generated.
@@ -151,7 +151,8 @@ def Do(
151151

152152
try:
153153
statistics_artifact.set_string_custom_property(
154-
STATS_DASHBOARD_LINK, dashboard_util.generate_stats_dashboard_link()
154+
STATS_DASHBOARD_LINK,
155+
stats_utils.generate_stats_dashboard_link(statistics_artifact),
155156
)
156157
except Exception as e: # pylint: disable=broad-except
157158
# log on failures to not bring down Statsgen jobs

Diff for: tfx/utils/stats_utils.py

+21
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,21 @@
1+
# Copyright 2024 Google LLC. All Rights Reserved.
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
"""stats_utils.
15+
16+
This module is the parity for internal implementation, not available in OSS.
17+
"""
18+
19+
20+
def generate_stats_dashboard_link(unused_statistics_artifact):
21+
return ''

0 commit comments

Comments
 (0)