Skip to content

Commit a9390bd

Browse files
tf-transform-teamtfx-copybara
authored andcommitted
Resolve unsoundness caught by pytype --strict-none-binding.
PiperOrigin-RevId: 555183577
1 parent d4c8b92 commit a9390bd

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensorflow_transform/beam/cached_impl_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -503,6 +503,8 @@ def _publish_rendered_dot_graph_file(self,
503503
dataset_keys, pcoll_cache_dict)
504504
def sort_value_node_values(cache_dict):
505505
result = []
506+
if cache_dict is None:
507+
return result
506508
for dataset_cache in cache_dict.values():
507509
result.extend(dataset_cache.values())
508510
return sorted(result, key=str)

0 commit comments

Comments
 (0)