Skip to content

Commit

Permalink
NA
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 713386174
  • Loading branch information
vkarampudi authored and Responsible ML Infra Team committed Jan 9, 2025
1 parent 504b489 commit 4a23760
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
8 changes: 4 additions & 4 deletions tensorboard_plugin/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,12 +43,12 @@ def select_constraint(default, nightly=None, git_master=None):

REQUIRED_PACKAGES = [
'protobuf>=3.20.3,<5',
'tensorboard>=2.15.2,<2.16.0',
'tensorflow>=2.15,<2.16',
'tensorboard>=2.16.2,<2.17.0',
'tensorflow>=2.16,<2.17',
'tensorflow-model-analysis'
+ select_constraint(
default='>=0.46,<0.47',
nightly='>=0.47.0.dev',
default='>=0.47,<0.48',
nightly='>=0.48.0.dev',
git_master='@git+https://github.com/tensorflow/model-analysis@master',
),
'werkzeug<2',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from __future__ import print_function

import os
from typing import Any
from typing import Any, Union

from absl import logging
from tensorboard_plugin_fairness_indicators import metadata
Expand Down Expand Up @@ -127,11 +127,11 @@ def convert_slicing_metrics_to_ui_input(
tfma.view.view_types.MetricsByOutputName,
]
],
slicing_column: str | None = None,
slicing_spec: tfma.slicer.slicer_lib.SingleSliceSpec | None = None,
slicing_column: Union[str, None] = None,
slicing_spec: Union[tfma.slicer.slicer_lib.SingleSliceSpec, None] = None,
output_name: str = '',
multi_class_key: str = '',
) -> list[dict[str, Any]] | None:
) -> Union[list[dict[str, Any]], None]:
"""Renders the Fairness Indicator view.
Args:
Expand Down

0 comments on commit 4a23760

Please sign in to comment.