Correct name of directory in tests

NVIDIA-Merlin · Jun 14, 2022 · 5738256 · 5738256
1 parent 604b232
commit 5738256
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/systems/fil/test_forest.py b/tests/unit/systems/fil/test_forest.py
@@ -87,7 +87,7 @@ def test_export(tmpdir):
 
     config_path = tmpdir / "2_forest" / "config.pbtxt"
     parsed_config = read_config(config_path)
-    assert parsed_config.name == "2_forest"
+    assert parsed_config.name == "2_predictforest"
     assert parsed_config.backend == "python"
 
     config_path = tmpdir / "2_fil" / "config.pbtxt"
@@ -126,9 +126,9 @@ def test_ensemble(tmpdir):
 
     triton_ens.export(tmpdir)
 
-    config_path = tmpdir / "1_forest" / "config.pbtxt"
+    config_path = tmpdir / "1_predictforest" / "config.pbtxt"
     parsed_config = read_config(config_path)
-    assert parsed_config.name == "1_forest"
+    assert parsed_config.name == "1_predictforest"
     assert parsed_config.backend == "python"
 
     config_path = tmpdir / "1_fil" / "config.pbtxt"