def __init__(self, headless=False):
self.logger = Logger("PureDriver").get_logger()
executable_path = os.path.join(
pathlib.Path(os.path.dirname(__file__)).parent,
"resources/chromedriver")
options = Options()
if headless:
options.headless = True
super().__init__(executable_path=executable_path, options=options)
self.implicitly_wait(wait_seconds)
def build_tensorboard(self):
"""Build a tensorboard logger."""
self.logger = Logger(self.cfg.dirs.log_dir)
def setUpClass(cls) -> None:
cls.logger = Logger(name="test").get_logger()
