Skip to content

Commit

Permalink
fix: Fixed issue with _keyspace population
Browse files Browse the repository at this point in the history
  • Loading branch information
Bhargav Dodla committed Jan 7, 2025
1 parent d8053b4 commit 4683447
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -352,6 +352,7 @@ def online_write_batch(
display progress.
"""
project = config.project
session: Session = self._get_session(config)

keyspace: str = self._keyspace
fqtable = CassandraOnlineStore._fq_table_name(keyspace, project, table)
Expand Down Expand Up @@ -383,6 +384,7 @@ def unroll_insertion_tuples() -> Iterable[Tuple[str, bytes, str, datetime]]:

self._write_rows_concurrently(
config,
session,
unroll_insertion_tuples(),
)
# correction for the last missing call to `progress`:
Expand Down Expand Up @@ -498,9 +500,9 @@ def _fq_table_name(keyspace: str, project: str, table: FeatureView) -> str:
def _write_rows_concurrently(
self,
config: RepoConfig,
session: Session,
batches: Iterable[BatchStatement],
):
session: Session = self._get_session(config)
futures = []
for batch in batches:
futures.append(session.execute_async(batch))
Expand Down

0 comments on commit 4683447

Please sign in to comment.