Skip to content

Commit 6aef937

Browse files
committed
Switch to create_table_and_save_to_sql
1 parent dbbf2b5 commit 6aef937

2 files changed

Lines changed: 5 additions & 9 deletions

File tree

src/graphnet/data/sqlite/__init__.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,7 @@
33
from graphnet.utilities.imports import has_torch_package
44

55
from .sqlite_dataconverter import SQLiteDataConverter
6-
from .sqlite_utilities import (
7-
run_sql_code,
8-
save_to_sql,
9-
create_table,
10-
create_table_and_save_to_sql,
11-
)
6+
from .sqlite_utilities import create_table_and_save_to_sql
127

138
if has_torch_package():
149
from .sqlite_dataset import SQLiteDataset

src/graphnet/pisa/fitting.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
from pisa.analysis.analysis import Analysis
2121
from pisa import ureg
2222

23-
from graphnet.data.sqlite import save_to_sql, create_table
23+
from graphnet.data.sqlite import create_table_and_save_to_sql
2424

2525
mpl.use("pdf")
2626
plt.rc("font", family="serif")
@@ -157,8 +157,9 @@ def fit_weights(
157157
results = results.append(data)
158158

159159
if add_to_database:
160-
create_table(results.columns, weight_name, self._database_path)
161-
save_to_sql(results, weight_name, self._database_path)
160+
create_table_and_save_to_sql(
161+
results.columns, weight_name, self._database_path
162+
)
162163
return results.sort_values("event_no").reset_index(drop=True)
163164

164165
def _make_config(

0 commit comments

Comments
 (0)