We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 37a5a38 commit 3ddce57Copy full SHA for 3ddce57
2 files changed
src/graphnet/data/parquet/__init__.py
@@ -1,2 +1,8 @@
1
+from graphnet.utilities.imports import has_torch_package
2
+
3
from .parquet_dataconverter import ParquetDataConverter
-from .parquet_dataset import ParquetDataset
4
5
+if has_torch_package():
6
+ from .parquet_dataset import ParquetDataset
7
8
+del has_torch_package
src/graphnet/data/sqlite/__init__.py
@@ -1,4 +1,10 @@
from .sqlite_dataconverter import SQLiteDataConverter
-from .sqlite_dataset import SQLiteDataset
-from .sqlite_dataset_perturbed import SQLiteDatasetPerturbed
from .sqlite_utilities import run_sql_code, save_to_sql
+ from .sqlite_dataset import SQLiteDataset
+ from .sqlite_dataset_perturbed import SQLiteDatasetPerturbed
9
10
0 commit comments