Lazy load torch in utils.basic

emdgroup · Mar 7, 2024 · 3401a4e · 3401a4e
1 parent c290908
commit 3401a4e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/baybe/utils/basic.py b/baybe/utils/basic.py
@@ -5,7 +5,6 @@
 from typing import Callable, Dict, Iterable, List, TypeVar
 
 import numpy as np
-import torch
 
 _C = TypeVar("_C", bound=type)
 _T = TypeVar("_T")
@@ -57,6 +56,8 @@ def set_random_seed(seed: int):
     Args:
         seed: The chosen global random seed.
     """
+    import torch
+
     torch.manual_seed(seed)
     random.seed(seed)
     np.random.seed(seed)