Skip to content

Commit ffb92bf

Browse files
committed
fix: type hint issue for some tf versions - removed it
1 parent 5e77983 commit ffb92bf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gradient_accumulator/accumulators.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ class GradientAccumulateOptimizer(tf.keras.optimizers.Optimizer):
113113
@typechecked
114114
def __init__(
115115
self,
116-
optimizer: types.Optimizer,
116+
optimizer, # : types.Optimizer, # having this results in TypeError -> expected OptimizerV2 or str, got dict instead
117117
accum_steps: types.TensorLike = 4,
118118
reduction: str = "MEAN",
119119
name: str = "GradientAccumulateOptimizer",

0 commit comments

Comments
 (0)