remove unused

kashif · Jan 21, 2025 · 761bc53 · 761bc53
1 parent 001b530
commit 761bc53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gluonts/torch/model/seg_diff/module.py b/src/gluonts/torch/model/seg_diff/module.py
@@ -22,7 +22,7 @@
 from gluonts.core.component import validated
 from gluonts.model import Input, InputSpec
 from gluonts.torch.scaler import StdScaler, MeanScaler, NOPScaler
-from gluonts.torch.util import unsqueeze_expand
+# from gluonts.torch.util import unsqueeze_expand
 
 from flow_matching.path import CondOTProbPath
 from flow_matching.solver import ODESolver