import unpack from typing_extensions

Author: inikishev

.github/workflows/tests.yml

@@ -16,7 +16,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     steps:
     - uses: actions/checkout@v4

src/torchzero/optim/experimental/experimental.py

@@ -1,5 +1,4 @@
-from collections.abc import Iterable
-from typing import Literal, Unpack
+from typing import Literal
 
 from ...modules import (
     LR,

src/torchzero/optim/first_order/cautious.py

@@ -1,4 +1,5 @@
-from typing import Literal, Unpack
+from typing import Literal
+
 
 from ...core import OptimizerModule
 from ...modules import Cautious, Adam, SGD, Lion, WeightDecay, LR

src/torchzero/optim/first_order/optimizers.py

@@ -1,5 +1,5 @@
 from collections.abc import Iterable
-from typing import Literal, Unpack
+from typing import Literal
 
 from ...modules import (
     LR,

src/torchzero/optim/zeroth_order/fdm.py

@@ -1,4 +1,4 @@
-from typing import Literal, Unpack
+from typing import Literal
 
 import torch
 

src/torchzero/optim/zeroth_order/rfdm.py

@@ -1,5 +1,3 @@
-from typing import Literal, Unpack
-
 import torch
 
 from ...modules import SGD, WrapClosure, LR

src/torchzero/tensorlist.py

@@ -13,8 +13,8 @@
 from collections.abc import Callable, Sequence, Iterable, Generator
 import math
 import operator
-from typing import Any, Literal, TypedDict, Unpack
-from typing_extensions import Self, TypeAlias
+from typing import Any, Literal, TypedDict
+from typing_extensions import Self, TypeAlias, Unpack
 
 import torch