Rename merlin.graph to merlin.dag (#22)

merlin/graph/base_operator.py → merlin/dag/base_operator.py

@@ -19,7 +19,7 @@
 from typing import Any, List, Union
 
 import merlin
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import ColumnSchema, Schema
 
 
@@ -248,7 +248,7 @@ def label(self) -> str:
         return self.__class__.__name__
 
     def create_node(self, selector):
-        return merlin.graph.Node(selector)
+        return merlin.dag.Node(selector)
 
     @property
     def supports(self) -> Supports:

merlin/graph/graph.py → merlin/dag/graph.py

@@ -17,7 +17,7 @@
 import logging
 from collections import deque
 
-from merlin.graph.node import (
+from merlin.dag.node import (
     Node,
     _combine_schemas,
     iter_nodes,

merlin/graph/node.py → merlin/dag/node.py

@@ -15,9 +15,9 @@
 #
 import collections.abc
 
-from merlin.graph.base_operator import BaseOperator
-from merlin.graph.ops import ConcatColumns, SelectionOp, SubsetColumns, SubtractionOp
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator
+from merlin.dag.ops import ConcatColumns, SelectionOp, SubsetColumns, SubtractionOp
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import Schema
 
 

merlin/graph/ops/concat_columns.py → merlin/dag/ops/concat_columns.py

@@ -15,8 +15,8 @@
 #
 
 from merlin.core.dispatch import DataFrameType
-from merlin.graph.base_operator import BaseOperator
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import Schema
 
 

merlin/graph/ops/identity.py → merlin/dag/ops/identity.py

@@ -15,8 +15,8 @@
 #
 
 from merlin.core.dispatch import DataFrameType
-from merlin.graph.base_operator import BaseOperator
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator
+from merlin.dag.selector import ColumnSelector
 
 
 class Identity(BaseOperator):

merlin/graph/ops/selection.py → merlin/dag/ops/selection.py

@@ -18,8 +18,8 @@
 import logging
 
 from merlin.core.dispatch import DataFrameType
-from merlin.graph.base_operator import BaseOperator
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import Schema
 
 LOG = logging.getLogger("SelectionOp")

merlin/graph/ops/subset_columns.py → merlin/dag/ops/subset_columns.py

@@ -15,8 +15,8 @@
 #
 
 from merlin.core.dispatch import DataFrameType
-from merlin.graph.base_operator import BaseOperator
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import Schema
 
 

merlin/graph/ops/subtraction.py → merlin/dag/ops/subtraction.py

@@ -16,8 +16,8 @@
 from __future__ import annotations
 
 from merlin.core.dispatch import DataFrameType
-from merlin.graph.base_operator import BaseOperator
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import Schema
 
 

merlin/graph/selector.py → merlin/dag/selector.py

@@ -46,7 +46,7 @@ def __init__(
         self._tags = tags if tags is not None else []
         self.subgroups = subgroups if subgroups is not None else []
 
-        if isinstance(self._names, merlin.graph.Node):
+        if isinstance(self._names, merlin.dag.Node):
             raise TypeError("ColumnSelectors can not contain Nodes")
 
         if isinstance(self._names, str):
@@ -59,7 +59,7 @@ def __init__(
         for name in self._names:
             if isinstance(name, str):
                 plain_names.append(name)
-            elif isinstance(name, merlin.graph.Node):
+            elif isinstance(name, merlin.dag.Node):
                 raise ValueError("ColumnSelectors can not contain Nodes")
             elif isinstance(name, ColumnSelector):
                 self.subgroups.append(name)
@@ -101,7 +101,7 @@ def _nested_check(self, nests=0):
     def __add__(self, other):
         if other is None:
             return self
-        elif isinstance(other, merlin.graph.Node):
+        elif isinstance(other, merlin.dag.Node):
             return other + self
         elif isinstance(other, ColumnSelector):
 
@@ -121,7 +121,7 @@ def __radd__(self, other):
         return self + other
 
     def __rshift__(self, operator):
-        if isinstance(operator, type) and issubclass(operator, merlin.graph.BaseOperator):
+        if isinstance(operator, type) and issubclass(operator, merlin.dag.BaseOperator):
             # handle case where an operator class is passed
             operator = operator()
 

tests/unit/graph/ops/test_selection.py → tests/unit/dag/ops/test_selection.py

@@ -15,8 +15,8 @@
 #
 import pytest
 
-from merlin.graph import ColumnSelector
-from merlin.graph.ops.selection import SelectionOp
+from merlin.dag import ColumnSelector
+from merlin.dag.ops.selection import SelectionOp
 from merlin.schema import ColumnSchema, Schema
 
 

tests/unit/graph/test_base_operator.py → tests/unit/dag/test_base_operator.py

@@ -15,9 +15,9 @@
 #
 import pytest
 
-from merlin.graph.base_operator import BaseOperator as Operator
-from merlin.graph.graph import Graph
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.base_operator import BaseOperator as Operator
+from merlin.dag.graph import Graph
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import Schema
 
 

tests/unit/graph/test_column_schemas.py → tests/unit/dag/test_column_schemas.py

@@ -18,7 +18,7 @@
 import numpy
 import pytest
 
-from merlin.graph.selector import ColumnSelector
+from merlin.dag.selector import ColumnSelector
 from merlin.schema import ColumnSchema, Schema
 from merlin.schema.io.tensorflow_metadata import TensorflowMetadata
 from merlin.schema.tags import Tags, TagSet

tests/unit/graph/test_column_selector.py → tests/unit/dag/test_column_selector.py

@@ -15,10 +15,10 @@
 #
 import pytest
 
-from merlin.graph import BaseOperator
-from merlin.graph.node import Node
-from merlin.graph.ops.selection import SelectionOp
-from merlin.graph.selector import ColumnSelector
+from merlin.dag import BaseOperator
+from merlin.dag.node import Node
+from merlin.dag.ops.selection import SelectionOp
+from merlin.dag.selector import ColumnSelector
 from merlin.schema.tags import Tags