Skip to content

Commit 3ddce57

Browse files
committed
Make available imports in graphnet.data.{sqlite,parquet} conditional on whether 'torch' is installed
1 parent 37a5a38 commit 3ddce57

2 files changed

Lines changed: 15 additions & 3 deletions

File tree

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,8 @@
1+
from graphnet.utilities.imports import has_torch_package
2+
13
from .parquet_dataconverter import ParquetDataConverter
2-
from .parquet_dataset import ParquetDataset
4+
5+
if has_torch_package():
6+
from .parquet_dataset import ParquetDataset
7+
8+
del has_torch_package
Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,10 @@
1+
from graphnet.utilities.imports import has_torch_package
2+
13
from .sqlite_dataconverter import SQLiteDataConverter
2-
from .sqlite_dataset import SQLiteDataset
3-
from .sqlite_dataset_perturbed import SQLiteDatasetPerturbed
44
from .sqlite_utilities import run_sql_code, save_to_sql
5+
6+
if has_torch_package():
7+
from .sqlite_dataset import SQLiteDataset
8+
from .sqlite_dataset_perturbed import SQLiteDatasetPerturbed
9+
10+
del has_torch_package

0 commit comments

Comments
 (0)