Firoj112 commited on
Commit
654b1c5
·
verified ·
1 Parent(s): 96526b8

Update tools/scrape_text.py

Browse files
Files changed (1) hide show
  1. tools/scrape_text.py +1 -3
tools/scrape_text.py CHANGED
@@ -15,9 +15,7 @@ class ScrapeTextTool(Tool):
15
  def __init__(self, driver):
16
  self.driver = driver
17
 
18
- def forward(self, **kwargs):
19
- selector = kwargs.get("selector", "p")
20
- extract_table = kwargs.get("extract_table", False)
21
  try:
22
  if extract_table:
23
  tables = self.driver.find_elements(By.CSS_SELECTOR, selector)
 
15
  def __init__(self, driver):
16
  self.driver = driver
17
 
18
+ def forward(self, selector="p", extract_table=False):
 
 
19
  try:
20
  if extract_table:
21
  tables = self.driver.find_elements(By.CSS_SELECTOR, selector)