marondeau commited on
Commit
c3ad77b
·
unverified ·
1 Parent(s): fb83544

Removed leftover and fixed docstring

Browse files
buster/documents/sqlite/documents.py CHANGED
@@ -34,7 +34,7 @@ class DocumentsDB(DocumentsManager):
34
 
35
  Example:
36
  >>> db = DocumentsDB("/path/to/the/db.db")
37
- >>> db.write_documents("source", df) # df is a DataFrame containing the documents from a given source, obtained e.g. by using buster.docparser.generate_embeddings
38
  >>> df = db.get_documents("source")
39
  """
40
 
@@ -150,7 +150,6 @@ class DocumentsDB(DocumentsManager):
150
  # Execute the SQL statement and fetch the results
151
  results = self.conn.execute("SELECT * FROM documents WHERE source = ?", (source,))
152
  rows = results.fetchall()
153
- print(rows[0])
154
 
155
  # Convert the results to a pandas DataFrame
156
  df = pd.DataFrame(rows, columns=[description[0] for description in results.description])
 
34
 
35
  Example:
36
  >>> db = DocumentsDB("/path/to/the/db.db")
37
+ >>> db.add("source", df) # df is a DataFrame containing the documents from a given source, obtained e.g. by using buster.docparser.generate_embeddings
38
  >>> df = db.get_documents("source")
39
  """
40
 
 
150
  # Execute the SQL statement and fetch the results
151
  results = self.conn.execute("SELECT * FROM documents WHERE source = ?", (source,))
152
  rows = results.fetchall()
 
153
 
154
  # Convert the results to a pandas DataFrame
155
  df = pd.DataFrame(rows, columns=[description[0] for description in results.description])