Merge pull request #157 from osqp/vb/nn_fail_explicit

failing as we should, instead of printing out an error
osqp · Aug 7, 2024 · dea9cc1 · dea9cc1
2 parents f088625 + 5cb4507
commit dea9cc1
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/osqp/nn/torch.py b/src/osqp/nn/torch.py
@@ -1,12 +1,9 @@
 import numpy as np
 import scipy.sparse as spa
 
-try:
-    import torch
-    from torch.nn import Module
-    from torch.autograd import Function
-except ImportError as e:
-    print(f'Import Error: {e}')
+import torch
+from torch.nn import Module
+from torch.autograd import Function
 from joblib import Parallel, delayed
 import multiprocessing