Rename graph building library files for consistency with the 'input_maker' module.

- graph_builder.py -> graph_builder_lib.py - graph_builder_test.py -> graph_builder_lib_test.py PiperOrigin-RevId: 272901940

Author: arjung

neural_structured_learning/tools/BUILD

@@ -29,7 +29,7 @@ py_library(
  srcs = ["__init__.py"],
  srcs_version = "PY2AND3",
  deps = [
- ":graph_builder",
+ ":graph_builder_lib",
  ":graph_utils",
  ":input_maker_lib",
  ],
@@ -56,8 +56,8 @@ py_test(
 )
 
 py_library(
- name = "graph_builder",
- srcs = ["graph_builder.py"],
+ name = "graph_builder_lib",
+ srcs = ["graph_builder_lib.py"],
  srcs_version = "PY2AND3",
  deps = [
  ":graph_utils",
@@ -69,11 +69,11 @@ py_library(
 )
 
 py_test(
- name = "graph_builder_test",
- srcs = ["graph_builder_test.py"],
+ name = "graph_builder_lib_test",
+ srcs = ["graph_builder_lib_test.py"],
  srcs_version = "PY2AND3",
  deps = [
- ":graph_builder",
+ ":graph_builder_lib",
  ":graph_utils",
  # package protobuf,
  # package absl/testing:absltest
@@ -86,7 +86,7 @@ py_binary(
  srcs = ["graph_builder_main.py"],
  python_version = "PY3",
  deps = [
- ":graph_builder",
+ ":graph_builder_lib",
  # package absl:app
  # package absl/flags
  # package tensorflow

neural_structured_learning/tools/__init__.py

@@ -1,6 +1,6 @@
 """Tools and APIs for preparing data for Neural Structured Learning."""
 
-from neural_structured_learning.tools.graph_builder import build_graph
+from neural_structured_learning.tools.graph_builder_lib import build_graph
 from neural_structured_learning.tools.graph_utils import add_edge
 from neural_structured_learning.tools.graph_utils import add_undirected_edges
 from neural_structured_learning.tools.graph_utils import read_tsv_graph

neural_structured_learning/tools/graph_builder.py renamed to neural_structured_learning/tools/graph_builder_lib.py

@@ -11,14 +11,14 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""Tests for neural_structured_learning.tools.graph_builder."""
+"""Tests for neural_structured_learning.tools.graph_builder_lib."""
 
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
 from absl.testing import absltest
-from neural_structured_learning.tools import graph_builder
+from neural_structured_learning.tools import graph_builder_lib
 from neural_structured_learning.tools import graph_utils
 import tensorflow as tf
 
@@ -82,9 +82,9 @@ def testGraphBuildingNoThresholding(self):
  embedding_path = self._create_embedding_file()
  self._write_embeddings(embedding_path)
  graph_path = self._create_graph_file()
- graph_builder.build_graph([embedding_path],
- graph_path,
- similarity_threshold=0)
+ graph_builder_lib.build_graph([embedding_path],
+  graph_path,
+  similarity_threshold=0)
  g_actual = graph_utils.read_tsv_graph(graph_path)
  self.assertDictEqual(
  g_actual, {
@@ -107,9 +107,9 @@ def testGraphBuildingWithThresholding(self):
  embedding_path = self._create_embedding_file()
  self._write_embeddings(embedding_path)
  graph_path = self._create_graph_file()
- graph_builder.build_graph([embedding_path],
- graph_path,
- similarity_threshold=0.51)
+ graph_builder_lib.build_graph([embedding_path],
+  graph_path,
+  similarity_threshold=0.51)
  g_actual = graph_utils.read_tsv_graph(graph_path)
  self.assertDictEqual(g_actual, {})
 

neural_structured_learning/tools/graph_builder_test.py renamed to neural_structured_learning/tools/graph_builder_lib_test.py

@@ -31,7 +31,7 @@
 
 from absl import app
 from absl import flags
-from neural_structured_learning.tools import graph_builder
+from neural_structured_learning.tools import graph_builder_lib
 import tensorflow as tf
 
 
@@ -44,8 +44,9 @@ def _main(argv):
  'Invalid number of arguments; expected 2 or more, got %d' %
  (len(argv) - 1))
 
- graph_builder.build_graph(argv[1:-1], argv[-1], flag.similarity_threshold,
- flag.id_feature_name, flag.embedding_feature_name)
+ graph_builder_lib.build_graph(argv[1:-1], argv[-1], flag.similarity_threshold,
+ flag.id_feature_name,
+ flag.embedding_feature_name)
 
 
 if __name__ == '__main__':