Merge pull request #63 from brandonwillard/fix-tflow-op-name-metatize

Do not attempt to metatize TFlowOpNames

Author: brandonwillard

requirements.txt

@@ -1,7 +1,7 @@
 scipy>=1.2.0
 Theano>=1.0.4
 tf-nightly-2.0-preview==2.0.0.dev20190606
-tfp-nightly>=0.7.0.dev20190320
+tfp-nightly==0.8.0.dev20190705
 pymc3>=3.6
 pymc4 @ git+https://github.com/pymc-devs/pymc4.git@master#egg=pymc4-0.0.1
 multipledispatch>=0.6.0

symbolic_pymc/tensorflow/meta.py

@@ -177,6 +177,9 @@ def __hash__(self):
         return hash((self._unique_name, self._in_idx))
 
 
+_metatize.add((TFlowOpName,), lambda x: x)
+
+
 def _metatize_tf_object(obj):
     try:
         obj = tf.convert_to_tensor(obj)

tests/tensorflow/test_meta.py

@@ -41,6 +41,8 @@ def test_op_names():
     # but not output numbers
     assert TFlowOpName('blah/add_1:0') != TFlowOpName('blah/add:1')
 
+    assert isinstance(mt(TFlowOpName('blah/add_1:0')), TFlowOpName)
+
 
 @pytest.mark.usefixtures("run_with_tensorflow")
 def test_meta_helper():