import order

Xilinx · Aug 25, 2024 · 4fac083 · 4fac083
1 parent f2924a5
commit 4fac083
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/proxy/parameter_quant.py b/src/brevitas/proxy/parameter_quant.py
@@ -8,6 +8,7 @@
 from warnings import warn
 
 import packaging.version
+import torch
 
 from brevitas import torch_version
 
@@ -16,7 +17,6 @@
 else:
     is_dynamo_compiling = torch._dynamo.is_compiling
 
-import torch
 from torch import Tensor
 import torch.nn as nn
 from typing_extensions import Protocol