Skip to content

Commit e4fb25d

Browse files
committed
Fix Roformer export symbol (#2199)
1 parent 23ac977 commit e4fb25d

File tree

4 files changed

+7
-7
lines changed

4 files changed

+7
-7
lines changed

Diff for: keras_hub/api/models/__init__.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -309,7 +309,7 @@
309309
)
310310
from keras_hub.src.models.roberta.roberta_tokenizer import RobertaTokenizer
311311
from keras_hub.src.models.roformer_v2.roformer_v2_backbone import (
312-
RoformerV2Backbone as RorformerV2Backbone,
312+
RoformerV2Backbone,
313313
)
314314
from keras_hub.src.models.roformer_v2.roformer_v2_masked_lm import (
315315
RoformerV2MaskedLM,
@@ -318,7 +318,7 @@
318318
RoformerV2MaskedLMPreprocessor,
319319
)
320320
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier import (
321-
RorformerV2TextClassifier,
321+
RoformerV2TextClassifier,
322322
)
323323
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier_preprocessor import (
324324
RoformerV2TextClassifierPreprocessor,

Diff for: keras_hub/src/models/roformer_v2/roformer_v2_backbone.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def roformer_kernel_initializer(stddev=0.02):
1616
return keras.initializers.TruncatedNormal(stddev=stddev)
1717

1818

19-
@keras_hub_export("keras_hub.models.RorformerV2Backbone")
19+
@keras_hub_export("keras_hub.models.RoformerV2Backbone")
2020
class RoformerV2Backbone(Backbone):
2121
"""A RoformerV2 encoder network.
2222

Diff for: keras_hub/src/models/roformer_v2/roformer_v2_text_classifier.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,8 @@
1010
)
1111

1212

13-
@keras_hub_export("keras_hub.models.RorformerV2TextClassifier")
14-
class RorformerV2TextClassifier(RobertaTextClassifier):
13+
@keras_hub_export("keras_hub.models.RoformerV2TextClassifier")
14+
class RoformerV2TextClassifier(RobertaTextClassifier):
1515
"""An end-to-end RoformerV2 model for classification tasks.
1616
1717
This model attaches a classification head to

Diff for: keras_hub/src/models/roformer_v2/roformer_v2_text_classifier_test.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
RoformerV2Backbone,
88
)
99
from keras_hub.src.models.roformer_v2.roformer_v2_text_classifier import (
10-
RorformerV2TextClassifier,
10+
RoformerV2TextClassifier,
1111
)
1212
from keras_hub.src.models.roformer_v2.roformer_v2_tokenizer import (
1313
RoformerV2Tokenizer,
@@ -49,7 +49,7 @@ def test_classifier_basics(self):
4949
if keras.__version__ < "3.6":
5050
self.skipTest("Failing on keras lower version")
5151
self.run_task_test(
52-
cls=RorformerV2TextClassifier,
52+
cls=RoformerV2TextClassifier,
5353
init_kwargs=self.init_kwargs,
5454
train_data=self.train_data,
5555
expected_output_shape=(2, 2),

0 commit comments

Comments
 (0)