migrated base runtime to core (#380)

* migrated base runtime to core

* add comments to init file

* change runtime imports

Author: jperez999

merlin/systems/dag/runtimes/__init__.py

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2022, NVIDIA CORPORATION.
+# Copyright (c) 2023, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,5 +14,3 @@
 # limitations under the License.
 #
 # flake8: noqa
-
-from .base_runtime import Runtime

merlin/systems/dag/runtimes/base_runtime.py

@@ -23,6 +23,7 @@
 from google.protobuf import text_format
 
 from merlin.dag import postorder_iter_nodes
+from merlin.dag.runtime import Runtime
 from merlin.systems.dag.ops import compute_dims
 from merlin.systems.dag.ops.compat import (
     cuml_ensemble,
@@ -32,7 +33,6 @@
     xgboost,
 )
 from merlin.systems.dag.ops.workflow import TransformWorkflow
-from merlin.systems.dag.runtimes import Runtime
 from merlin.systems.dag.runtimes.triton.ops.operator import TritonOperator, add_model_param
 from merlin.systems.dag.runtimes.triton.ops.workflow import TransformWorkflowTriton
 

merlin/systems/dag/runtimes/triton/runtime.py

@@ -3,11 +3,11 @@
 import pytest
 
 from merlin.dag import ColumnSelector
+from merlin.dag.runtime import Runtime
 from merlin.dtypes.shape import Shape
 from merlin.schema import ColumnSchema, Schema
 from merlin.systems.dag.ensemble import Ensemble
 from merlin.systems.dag.ops.fil import PredictForest
-from merlin.systems.dag.runtimes.base_runtime import Runtime
 from merlin.table import TensorTable
 
 sklearn_datasets = pytest.importorskip("sklearn.datasets")

tests/unit/systems/dag/runtimes/local/ops/fil/test_lightgbm.py

@@ -3,11 +3,11 @@
 import pytest
 
 from merlin.dag import ColumnSelector
+from merlin.dag.runtime import Runtime
 from merlin.dtypes.shape import Shape
 from merlin.schema import ColumnSchema, Schema
 from merlin.systems.dag.ensemble import Ensemble
 from merlin.systems.dag.ops.fil import PredictForest
-from merlin.systems.dag.runtimes.base_runtime import Runtime
 from merlin.table import TensorTable
 
 sklearn_datasets = pytest.importorskip("sklearn.datasets")

tests/unit/systems/dag/runtimes/local/ops/fil/test_sklearn.py

@@ -4,11 +4,11 @@
 
 from merlin.core.compat import HAS_GPU
 from merlin.dag import ColumnSelector
+from merlin.dag.runtime import Runtime
 from merlin.dtypes.shape import Shape
 from merlin.schema import ColumnSchema, Schema
 from merlin.systems.dag.ensemble import Ensemble
 from merlin.systems.dag.ops.fil import PredictForest
-from merlin.systems.dag.runtimes.base_runtime import Runtime
 from merlin.table import TensorTable
 
 sklearn_datasets = pytest.importorskip("sklearn.datasets")

tests/unit/systems/dag/runtimes/local/ops/fil/test_xgboost.py

@@ -16,7 +16,7 @@
 import numpy as np
 import pytest
 
-from merlin.systems.dag.runtimes.base_runtime import Runtime
+from merlin.dag.runtime import Runtime
 from merlin.table import TensorTable
 from nvtabular import Workflow
 from nvtabular import ops as wf_ops

tests/unit/systems/dag/runtimes/local/ops/nvtabular/test_ensemble.py

@@ -17,8 +17,8 @@
 import pytest
 
 from merlin.dag import ColumnSelector
+from merlin.dag.runtime import Runtime
 from merlin.schema import Tags
-from merlin.systems.dag.runtimes.base_runtime import Runtime
 from nvtabular import Workflow
 from nvtabular import ops as wf_ops
 

tests/unit/systems/dag/runtimes/local/ops/tensorflow/test_ensemble.py

@@ -18,9 +18,9 @@
 import numpy as np
 import pytest
 
+from merlin.dag.runtime import Runtime
 from merlin.schema import ColumnSchema, Schema
 from merlin.systems.dag.ensemble import Ensemble
-from merlin.systems.dag.runtimes.base_runtime import Runtime
 from merlin.table import TensorTable
 
 torch = pytest.importorskip("torch")