Fixes failing tests following refactor

MoseleyS · Sep 21, 2023 · 4a55f1f · 4a55f1f
1 parent 240ed3d
commit 4a55f1f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/improver_tests/acceptance/test_categorical.py b/improver_tests/acceptance/test_categorical.py
@@ -214,12 +214,12 @@ def test_without_optional_input(tmp_path):
         ("bad_wx_decision_tree.json", "Unreachable node 'unreachable'"),
     ),
 )
-def test_trees(wxtree, expected):
+def test_trees(decision_tree, expected):
     """Test the check-tree option"""
     kgo_dir = acc.kgo_root() / "wxcode"
     args = [
         "--decision-tree",
-        kgo_dir / wxtree,
+        kgo_dir / decision_tree,
         "--check-tree",
         "--target-period",
         "3600",

diff --git a/improver_tests/developer_tools/conftest.py b/improver_tests/developer_tools/conftest.py
@@ -45,7 +45,7 @@
     set_up_probability_cube,
     set_up_variable_cube,
 )
-from improver.wxcode.utilities import weather_code_attributes
+from improver.categorical.utilities import weather_code_attributes
 
 
 def _update_blended_time_coords(cube):