Make _cohort.py module a public module (#1253)

* Make _cohort.py a public module Signed-off-by: Gaurav Gupta <gaugup@microsoft.com> * Add missing file Signed-off-by: Gaurav Gupta <gaugup@microsoft.com>
microsoft · Mar 8, 2022 · b47627e · b47627e
1 parent 1b2bc5e
commit b47627e
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/raiwidgets/raiwidgets/_cohort.py → raiwidgets/raiwidgets/cohort.py b/raiwidgets/raiwidgets/_cohort.py → raiwidgets/raiwidgets/cohort.py
diff --git a/raiwidgets/raiwidgets/responsibleai_dashboard_input.py b/raiwidgets/raiwidgets/responsibleai_dashboard_input.py
@@ -12,7 +12,7 @@
 from responsibleai._input_processing import _convert_to_list
 from responsibleai.exceptions import UserConfigValidationException
 
-from ._cohort import Cohort
+from .cohort import Cohort
 from .constants import ErrorMessages
 from .error_handling import _format_exception
 from .interfaces import WidgetRequestResponseConstants

diff --git a/raiwidgets/tests/test_cohort.py b/raiwidgets/tests/test_cohort.py
@@ -6,9 +6,9 @@
 import pandas as pd
 import pytest
 
-from raiwidgets._cohort import (ClassificationOutcomes, Cohort, CohortFilter,
-                                CohortFilterMethods,
-                                cohort_filter_json_converter)
+from raiwidgets.cohort import (ClassificationOutcomes, Cohort, CohortFilter,
+                               CohortFilterMethods,
+                               cohort_filter_json_converter)
 from responsibleai.exceptions import UserConfigValidationException
 
 

diff --git a/raiwidgets/tests/test_responsibleai_dashboard.py b/raiwidgets/tests/test_responsibleai_dashboard.py
@@ -6,7 +6,7 @@
 import pytest
 
 from raiwidgets import ResponsibleAIDashboard
-from raiwidgets._cohort import Cohort, CohortFilter, CohortFilterMethods
+from raiwidgets.cohort import Cohort, CohortFilter, CohortFilterMethods
 from responsibleai._interfaces import (CausalData, CounterfactualData, Dataset,
                                        ErrorAnalysisData, ModelExplanationData)
 from responsibleai.exceptions import UserConfigValidationException