Merge pull request #1828 from chrism0dwk:chrism0dwk-patch-1

PiperOrigin-RevId: 661311249

tensorflow_probability/python/internal/structural_tuple.py

@@ -142,7 +142,9 @@ def __getnewargs__(self):
       """Return self as a plain tuple.  Used by copy and pickle."""
       return tuple(self)
 
-    def __getattr__(self, attr):
+    def __getattribute__(self, attr):
+      if attr.startswith('_'):
+        return super().__getattribute__(attr)
       attr_idx = self._field_name_to_index.get(attr)
       if attr_idx is None:
         raise AttributeError('StructTuple has no attribute {}'.format(attr))

tensorflow_probability/python/internal/structural_tuple_test.py

@@ -100,6 +100,10 @@ def foo(a, b):
 
     self.assertEqual(3, nest_util.call_fn(foo, t(1, 2)))
 
+  def testTupleMethods(self):
+    t = structural_tuple.structtuple(['count', 'a', 'b'])
+    self.assertEqual(t(1, 2, 3).count, 1)
+
   def testMoreThan255Fields(self):
     num_fields = 1000
     t = structural_tuple.structtuple(