Remove check_types from nest.map_up_to.

This should not change the behavior, but eliminates warnings (the warning tells you that the argument does nothing). See #1523 for a description of the problem.

PiperOrigin-RevId: 451173112

Author: ColCarroll

tensorflow_probability/python/internal/backend/numpy/nest.py

@@ -327,8 +327,9 @@ def map_structure_with_tuple_paths_up_to(shallow_structure, func, *structures,
   if not structures:
     raise ValueError('Cannot map over no sequences')
 
+  # Internal `tree` does not accept check_types here; see b/198436438.
   check_types = kwargs.get('check_types', True)
-#   kwargs.pop('check_types', None)  # DisableOnExport
+  kwargs.pop('check_types', None)
 
   if expand_composites:
     raise NotImplementedError(