Skip to content

Commit

Permalink
PL Support on raiwidgets (#2500)
Browse files Browse the repository at this point in the history
* pl support

* python lint fixes

* Updated rai_core_flask requirement
  • Loading branch information
Advitya17 authored Jan 24, 2024
1 parent 01cb9d5 commit 9f7c545
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 3 deletions.
5 changes: 4 additions & 1 deletion raiwidgets/raiwidgets/dashboard.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,14 +61,17 @@ def __init__(self, *,
port,
locale,
no_inline_dashboard=False,
is_private_link=False,
**kwargs):
"""Initialize the dashboard."""

if model_data is None or type is None:
raise ValueError("Required parameters not provided")

try:
self._service = FlaskHelper(ip=public_ip, port=port)
self._service = FlaskHelper(ip=public_ip,
port=port,
is_private_link=is_private_link)
except Exception as e:
self._service = None
raise e
Expand Down
7 changes: 6 additions & 1 deletion raiwidgets/raiwidgets/responsibleai_dashboard.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,10 +29,14 @@ class ResponsibleAIDashboard(Dashboard):
:param cohort_list:
List of cohorts defined by the user for the dashboard.
:type cohort_list: List[Cohort]
:param is_private_link: If the dashboard environment is
a private link AML workspace.
:type is_private_link: bool
"""
def __init__(self, analysis: RAIInsights,
public_ip=None, port=None, locale=None,
cohort_list=None, **kwargs):
cohort_list=None, is_private_link=False,
**kwargs):
self.input = ResponsibleAIDashboardInput(
analysis, cohort_list=cohort_list)

Expand All @@ -43,6 +47,7 @@ def __init__(self, analysis: RAIInsights,
port=port,
locale=locale,
no_inline_dashboard=True,
is_private_link=is_private_link,
**kwargs)

def predict():
Expand Down
2 changes: 1 addition & 1 deletion raiwidgets/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
numpy>=1.17.2,<=1.26.2
pandas>=0.25.1,<2.0.0
scipy>=1.4.1
rai-core-flask==0.7.2
rai-core-flask==0.7.3
itsdangerous<=2.1.2
scikit-learn>=0.22.1
lightgbm>=2.0.11
Expand Down

0 comments on commit 9f7c545

Please sign in to comment.