Skip to content

Commit bac4b98

Browse files
committed
Adds hotfix for generator in load_csv.py
1 parent 68ce9f2 commit bac4b98

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

‎src/load_csv.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from dialog.models import CompanyContent
99
from dialog.models.db import get_session
1010

11-
session = get_session()
11+
session = next(get_session())
1212

1313
def load_csv_and_generate_embeddings(path, cleardb=False, embed_columns=("content",)):
1414
df = pd.read_csv(path)

0 commit comments

Comments
 (0)