Skip to content

Commit

Permalink
rollback import branches
Browse files Browse the repository at this point in the history
  • Loading branch information
chunnienc committed Dec 16, 2024
1 parent 7ce8b8d commit 7b842fb
Showing 2 changed files with 2 additions and 12 deletions.
7 changes: 1 addition & 6 deletions experimental/torch_xla2/torch_xla2/ops/op_base.py
Original file line number Diff line number Diff line change
@@ -7,12 +7,7 @@
from torch_xla2 import types
import sys

if sys.version_info < (3, 10):
from typing_extensions import ParamSpec, Concatenate
else:
from typing import ParamSpec, Concatenate

from typing import Callable, Optional
from typing import Callable, Optional, ParamSpec, Concatenate


class InplaceOp:
7 changes: 1 addition & 6 deletions experimental/torch_xla2/torch_xla2/types.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,9 @@
from typing import Callable, Any, Union
from typing import Callable, Any, Union, ParamSpec, TypeAlias
import torch
import jax
import jax.numpy as jnp
import sys

if sys.version_info < (3, 10):
from typing_extensions import ParamSpec, TypeAlias
else:
from typing import ParamSpec, TypeAlias

P = ParamSpec('P')

TorchValue: TypeAlias = Union[torch.Tensor, torch.dtype, 'TorchCallable', Any]

0 comments on commit 7b842fb

Please sign in to comment.