Correct incorrect import

golmschenk · Aug 15, 2024 · f90edf3 · f90edf3
1 parent 36ee270
commit f90edf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/qusi/internal/infinite_datasets_test_session.py b/src/qusi/internal/infinite_datasets_test_session.py
@@ -1,7 +1,7 @@
 from torch.nn import Module
 from torch.types import Device
 from torch.utils.data import DataLoader
-from wandb.wandb_torch import torch
+import torch
 
 from qusi.internal.light_curve_dataset import LightCurveDataset