From c85c115ba98bd0cd38c45f830489f3336bef8882 Mon Sep 17 00:00:00 2001
From: shoubhik <shoubhikbhatti@gmail.com>
Date: Tue, 17 Sep 2019 16:39:34 -0700
Subject: [PATCH] Adding support to check if an attribute is present or not
 without having to get the value (#3957)

* Adding support to check if an attribute is present or not without having to get the value.

* - Renaming the method to more appropriate name.
---
 python/tvm/relay/frontend/common.py  | 14 ++++++++++++
 tests/python/frontend/test_common.py | 32 ++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+)
 create mode 100644 tests/python/frontend/test_common.py

diff --git a/python/tvm/relay/frontend/common.py b/python/tvm/relay/frontend/common.py
index 81c4ce50154c..dbb4197e0c63 100644
--- a/python/tvm/relay/frontend/common.py
+++ b/python/tvm/relay/frontend/common.py
@@ -41,6 +41,20 @@ class StrAttrsDict(object):
     def __init__(self, attrs):
         self.attrs = attrs
 
+    def has_attr(self, key):
+        """Checks if a attribute is present in the map.
+
+        Parameters
+        ----------
+        key : str
+            The attribute key
+
+        Returns
+        -------
+        bool : True if the key is present in the attributes else false.
+        """
+        return key in self.attrs
+
     def get_float(self, key, default=RequiredAttr()):
         """Get float attribute
 
diff --git a/tests/python/frontend/test_common.py b/tests/python/frontend/test_common.py
new file mode 100644
index 000000000000..8ce858ec45a2
--- /dev/null
+++ b/tests/python/frontend/test_common.py
@@ -0,0 +1,32 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+from tvm.relay.frontend.common import StrAttrsDict
+
+
+def test_key_is_present():
+    attrs = StrAttrsDict({"a": 1})
+    assert attrs.has_attr("a")
+
+
+def test_key_is_not_present():
+    attrs = StrAttrsDict({"a": 1})
+    assert not attrs.has_attr("b")
+
+
+if __name__ == '__main__':
+    test_key_is_present()
+    test_key_is_present()