Skip to content

Commit

Permalink
remove unused class (#3123)
Browse files Browse the repository at this point in the history
### Changes

Removed TensorBackendType.
### Reason for changes

Got TensorBackendType and TensorBackend with the same definition
  • Loading branch information
AlexanderDokuchaev authored Nov 29, 2024
1 parent 9c5220a commit 7c64c3d
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 13 deletions.
2 changes: 1 addition & 1 deletion nncf/tensor/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.

from nncf.tensor.definitions import TensorBackendType as TensorBackendType
from nncf.tensor.definitions import TensorBackend as TensorBackend
from nncf.tensor.definitions import TensorDataType as TensorDataType
from nncf.tensor.definitions import TensorDeviceType as TensorDeviceType
from nncf.tensor.tensor import Tensor as Tensor
Expand Down
15 changes: 3 additions & 12 deletions nncf/tensor/definitions.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,13 @@
from enum import auto


class TensorBackendType(Enum):
class TensorBackend(Enum):
"""
Enum representing the different tensor backends.
"""

NUMPY = auto()
TORCH = auto()
numpy = auto()
torch = auto()


class TensorDataType(Enum):
Expand Down Expand Up @@ -53,15 +53,6 @@ class TensorDeviceType(Enum):
GPU = auto()


class TensorBackend(Enum):
"""
Enum representing the different tensor backends.
"""

numpy = auto()
torch = auto()


@dataclass
class TypeInfo:
"""
Expand Down

0 comments on commit 7c64c3d

Please sign in to comment.