Skip to content

Commit ca2baa7

Browse files
committed
Catch missing alienpy
1 parent e441292 commit ca2baa7

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

machine_learning_hep/utils/dl_train.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,11 @@
77
from pathlib import PurePosixPath
88

99
import requests # pylint: disable=import-error
10-
from alienpy import alien, xrd_core
10+
11+
try:
12+
from alienpy import alien, xrd_core
13+
except ImportError:
14+
print("Could not import alien, install with pip install alienpy")
1115

1216

1317
def get_train_spec(train_id: int):

0 commit comments

Comments
 (0)