From f90edf3e23e8ab76d45165cff3671898d4094bfe Mon Sep 17 00:00:00 2001 From: golmschenk Date: Thu, 15 Aug 2024 13:43:59 -0400 Subject: [PATCH] Correct incorrect import --- src/qusi/internal/infinite_datasets_test_session.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/qusi/internal/infinite_datasets_test_session.py b/src/qusi/internal/infinite_datasets_test_session.py index 5abddb7..f033fa7 100644 --- 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