Skip to content

Commit 97a5555

Browse files
remove annotations import (#31)
1 parent d61790b commit 97a5555

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

ppq/quantization/algorithm/training.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,4 @@
11
# TODO move training logic to here.
2-
from __future__ import annotations
3-
42
import random
53
from math import sqrt
64
from random import randint
@@ -650,7 +648,7 @@ def __init__(self,
650648
self._masters = []
651649

652650
@ property
653-
def masters(self) -> List[StraightThroughEstimateDelegator]:
651+
def masters(self):
654652
return self._masters
655653

656654
@ masters.setter
@@ -675,6 +673,7 @@ def finalize(self) -> None:
675673
# bias
676674
scale = self.scale_multiplier
677675
for delegator in self.masters:
676+
assert isinstance(delegator, StraightThroughEstimateDelegator)
678677
scale = scale * delegator.scale.data.abs()
679678
self.config.scale = scale
680679

0 commit comments

Comments
 (0)