Internal change

PiperOrigin-RevId: 303925062
minmingogogo · Mar 31, 2020 · ba508d9 · ba508d9
1 parent 2e96d5d
commit ba508d9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/tensorflow_hub/feature_column.py b/tensorflow_hub/feature_column.py
@@ -354,7 +354,9 @@ def parents(self):
   def name(self):
     """Returns string. Used for variable_scope and naming."""
     if not hasattr(self, "_name"):
-      self._name = "{}_hub_module_embedding".format(self.key)
+      key_name = self.key if isinstance(self.key,
+                                        six.string_types) else self.key.name
+      self._name = "{}_hub_module_embedding".format(key_name)
     return self._name
 
   def create_state(self, state_manager):

diff --git a/tensorflow_hub/feature_column_test.py b/tensorflow_hub/feature_column_test.py
@@ -407,6 +407,11 @@ def testConfig(self):
       image_column = hub.image_embedding_column("image", self.spec)
       config = image_column.get_config()
 
+  def testName(self):
+    image_column = hub.image_embedding_column(
+        tf.feature_column.numeric_column("image"), self.spec)
+    self.assertEqual("image_hub_module_embedding", image_column.name)
+
 
 class SparseTextEmbeddingColumnTest(tf.test.TestCase):