Remove runtime parameter from implicit ensemble tests (#358)

* Remove runtime parameter from implicit test ensemble

* Remove unused TritonExecutorRuntime

Author: oliverholworthy

tests/unit/systems/ops/implicit/test_op.py

@@ -25,7 +25,6 @@
 from merlin.schema import ColumnSchema, Schema
 from merlin.systems.dag.ensemble import Ensemble
 from merlin.systems.dag.ops.implicit import PredictImplicit
-from merlin.systems.dag.runtimes.triton import TritonExecutorRuntime
 from merlin.systems.triton.utils import run_triton_server
 
 TRITON_SERVER_PATH = shutil.which("tritonserver")
@@ -35,7 +34,6 @@
 
 
 @pytest.mark.skipif(not TRITON_SERVER_PATH, reason="triton server not found")
-@pytest.mark.parametrize("runtime", [None, TritonExecutorRuntime()])
 @pytest.mark.parametrize(
     "model_cls",
     [
@@ -44,7 +42,7 @@
         implicit.lmf.LogisticMatrixFactorization,
     ],
 )
-def test_ensemble(model_cls, runtime, tmpdir):
+def test_ensemble(model_cls, tmpdir):
     model = model_cls()
     n = 100
     user_items = csr_matrix(np.random.choice([0, 1], size=n * n, p=[0.9, 0.1]).reshape(n, n))
@@ -64,7 +62,7 @@ def test_ensemble(model_cls, runtime, tmpdir):
     triton_chain = input_schema.column_names >> implicit_op
 
     triton_ens = Ensemble(triton_chain, input_schema)
-    ensemble_config, _ = triton_ens.export(tmpdir, runtime=runtime)
+    ensemble_config, _ = triton_ens.export(tmpdir)
 
     input_user_id = np.array([[0], [1]], dtype=np.int64)
     inputs = [