Make method public since it's used in tests

dbt-labs · Apr 17, 2024 · 352771f · 352771f
1 parent 21d8572
commit 352771f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/metricflow/model/semantics/linkable_spec_resolver.py b/metricflow/model/semantics/linkable_spec_resolver.py
@@ -899,9 +899,7 @@ def _get_joined_elements(self, measure_semantic_model: SemanticModel) -> Linkabl
 
         return LinkableElementSet.merge_by_path_key((single_hop_elements, multi_hop_elements))
 
-    def _get_all_linkable_metrics_for_semantic_model(
-        self, starting_semantic_model: SemanticModel
-    ) -> LinkableElementSet:
+    def get_all_linkable_metrics_for_semantic_model(self, starting_semantic_model: SemanticModel) -> LinkableElementSet:
         """Get all metrics that can be joined to this semantic model using up to the max allowed join hops."""
         linkable_metrics = LinkableElementSet()
 
@@ -946,7 +944,7 @@ def _get_linkable_element_set_for_measure(
         elements_in_semantic_model = self._get_elements_in_semantic_model(measure_semantic_model)
         metric_time_elements = self._get_metric_time_elements(measure_reference)
         joined_elements = self._get_joined_elements(measure_semantic_model)
-        linkable_metrics = self._get_all_linkable_metrics_for_semantic_model(measure_semantic_model)
+        linkable_metrics = self.get_all_linkable_metrics_for_semantic_model(measure_semantic_model)
 
         return LinkableElementSet.merge_by_path_key(
             (

diff --git a/tests/model/test_semantic_model_container.py b/tests/model/test_semantic_model_container.py
@@ -211,7 +211,7 @@ def test_linkable_metrics_for_semantic_model(  # noqa: D103
         request=request,
         mf_test_configuration=mf_test_configuration,
         set_id="result0",
-        linkable_element_set=metric_lookup._linkable_spec_resolver._get_all_linkable_metrics_for_semantic_model(
+        linkable_element_set=metric_lookup._linkable_spec_resolver.get_all_linkable_metrics_for_semantic_model(
             measure_semantic_model
         ),
     )