SeleniumHQ
diff --git a/‎py/selenium/webdriver/chrome/options.py‎
Lines changed: 20 additions & 0 deletions b/‎py/selenium/webdriver/chrome/options.py‎
Lines changed: 20 additions & 0 deletions
@@ -151,6 +151,26 @@ def add_experimental_option(self, name, value):
         """
         self._experimental_options[name] = value
 
+    @property
+    def headless(self):
+        """
+        Returns whether or not the headless argument is set
+        """
+        return '--headless' in self._arguments
+
+    def set_headless(self, headless=True):
+        """
+        Sets the headless argument
+
+        Args:
+          headless: boolean value indicating to set the headless option
+        """
+        args = {'--headless', '--disable-gpu'}
+        if headless:
+            self._arguments.extend(args)
+        else:
+            self._arguments = list(set(self._arguments) - args)
+
     def to_capabilities(self):
         """
             Creates a capabilities with all the options that have been set and