From 3870da2059ddfc27d426135d0cbc265f3a234c95 Mon Sep 17 00:00:00 2001
From: janEbert <janpublicebert@posteo.net>
Date: Thu, 2 Nov 2023 09:55:42 +0100
Subject: [PATCH] Rename `typing` module to `typing_utils`
So we do not have clashes with the built-in module.
---
lee_transformers/layers/adaptive_computation_time.py | 2 +-
lee_transformers/layers/glu_layers.py | 2 +-
lee_transformers/layers/resampling.py | 2 +-
lee_transformers/models/hierarchical_transformer.py | 2 +-
lee_transformers/models/lpe_transformer.py | 2 +-
lee_transformers/models/rpe_transformer.py | 2 +-
lee_transformers/models/transformer.py | 2 +-
lee_transformers/{typing.py => typing_utils.py} | 0
8 files changed, 7 insertions(+), 7 deletions(-)
rename lee_transformers/{typing.py => typing_utils.py} (100%)
diff --git a/lee_transformers/layers/adaptive_computation_time.py b/lee_transformers/layers/adaptive_computation_time.py
index 8209635..d0b5ab6 100644
--- a/lee_transformers/layers/adaptive_computation_time.py
+++ b/lee_transformers/layers/adaptive_computation_time.py
@@ -9,7 +9,7 @@ from typing import Any, cast, Optional, Tuple
import torch
import torch as th
-from ..typing import ActivationFn
+from ..typing_utils import ActivationFn
class AdaptiveComputationTime(th.nn.Module):
diff --git a/lee_transformers/layers/glu_layers.py b/lee_transformers/layers/glu_layers.py
index 46e2c1e..d6ee6dc 100644
--- a/lee_transformers/layers/glu_layers.py
+++ b/lee_transformers/layers/glu_layers.py
@@ -3,7 +3,7 @@ from typing import Optional
import torch
import torch as th
-from ..typing import ActivationFn
+from ..typing_utils import ActivationFn
class GLULayer(th.nn.Module):
diff --git a/lee_transformers/layers/resampling.py b/lee_transformers/layers/resampling.py
index 85f6dda..7565d04 100644
--- a/lee_transformers/layers/resampling.py
+++ b/lee_transformers/layers/resampling.py
@@ -6,7 +6,7 @@ import torch as th
from .glu_layers import GLULayer
from .rms_norm import RMSNorm
-from ..typing import ActivationFn
+from ..typing_utils import ActivationFn
def shift_right(
diff --git a/lee_transformers/models/hierarchical_transformer.py b/lee_transformers/models/hierarchical_transformer.py
index 06d3827..b1b45c3 100644
--- a/lee_transformers/models/hierarchical_transformer.py
+++ b/lee_transformers/models/hierarchical_transformer.py
@@ -32,7 +32,7 @@ from ..layers import (
RMSNorm,
shift_right,
)
-from ..typing import ActivationFn
+from ..typing_utils import ActivationFn
OptionalMaskList = Union[List[torch.Tensor], List[None], bool, None]
diff --git a/lee_transformers/models/lpe_transformer.py b/lee_transformers/models/lpe_transformer.py
index bbf0b4c..d59ae98 100644
--- a/lee_transformers/models/lpe_transformer.py
+++ b/lee_transformers/models/lpe_transformer.py
@@ -7,7 +7,7 @@ import torch as th
from . import common
from ..layers import GLULayer, RMSNorm
-from ..typing import ActivationFn, Decoder, DecoderLayer
+from ..typing_utils import ActivationFn, Decoder, DecoderLayer
class LPETransformer(th.nn.Module):
diff --git a/lee_transformers/models/rpe_transformer.py b/lee_transformers/models/rpe_transformer.py
index ac90f56..84bf2c9 100644
--- a/lee_transformers/models/rpe_transformer.py
+++ b/lee_transformers/models/rpe_transformer.py
@@ -7,7 +7,7 @@ import torch as th
from . import common
from ..layers import GLULayer, RMSNorm, RPEMultiheadAttention
-from ..typing import ActivationFn, Decoder, DecoderLayer
+from ..typing_utils import ActivationFn, Decoder, DecoderLayer
# Encoder self-attention block (private method of
diff --git a/lee_transformers/models/transformer.py b/lee_transformers/models/transformer.py
index 9053b0b..f7651f1 100644
--- a/lee_transformers/models/transformer.py
+++ b/lee_transformers/models/transformer.py
@@ -7,7 +7,7 @@ import torch as th
from . import common
from ..layers import GLULayer, RMSNorm
-from ..typing import ActivationFn, Decoder, DecoderLayer
+from ..typing_utils import ActivationFn, Decoder, DecoderLayer
class Transformer(th.nn.Module):
diff --git a/lee_transformers/typing.py b/lee_transformers/typing_utils.py
similarity index 100%
rename from lee_transformers/typing.py
rename to lee_transformers/typing_utils.py
--
GitLab