Update mlir/utils/generate-test-checks.py

Co-authored-by: Mehdi Amini <[email protected]>

Author: tdanyluk

mlir/utils/generate-test-checks.py

@@ -148,9 +148,7 @@ def generate_name(self, source_attribute_name):
     # has been generated for the given attribute yet, None is returned.
     def get_name(self, source_attribute_name):
         return (
-            self.map[source_attribute_name]
-            if source_attribute_name in self.map
-            else None
+            self.map.get(source_attribute_name)
         )
 
 # Return the number of SSA results in a line of type