Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Commit 5e2dfa7

Browse files
dbogunowiczbogunowicz@arrival.com
andauthored
Remove typing.Final (#1205)
Co-authored-by: bogunowicz@arrival.com <bogunowicz@arrival.com>
1 parent 68f2a2f commit 5e2dfa7

File tree

1 file changed

+2
-2
lines changed
  • src/sparseml/transformers/sparsification

1 file changed

+2
-2
lines changed

src/sparseml/transformers/sparsification/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
import warnings
2525
from contextlib import suppress
2626
from dataclasses import asdict
27-
from typing import Any, Dict, Final, List, Optional, Tuple, Union
27+
from typing import Any, Dict, List, Optional, Tuple, Union
2828

2929
import datasets
3030
import numpy
@@ -761,7 +761,7 @@ def _get_fake_dataloader(
761761
# Rearrange inputs' keys to match those defined by model foward func, which
762762
# seem to define how the order of inputs is determined in the exported model
763763
forward_args_spec = inspect.getfullargspec(self.model.__class__.forward)
764-
synthetic_input: Final = self._get_fake_input(
764+
synthetic_input = self._get_fake_input(
765765
forward_func_input_keys=forward_args_spec.args,
766766
tokenizer=tokenizer,
767767
)

0 commit comments

Comments
 (0)