Hi I was using vscode for python selenium.
I used this code
from selenium import webdriver
def getHeadlessDriver(): global dr