Export video swin APIs (#2428)

* Export APIs * generate API
keras-team · Apr 29, 2024 · c54f4a8 · c54f4a8
1 parent fcf9576
commit c54f4a8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/keras_cv/api/models/__init__.py b/keras_cv/api/models/__init__.py
@@ -178,6 +178,15 @@
     ResNetV2Backbone,
 )
 from keras_cv.src.models.backbones.vgg16.vgg16_backbone import VGG16Backbone
+from keras_cv.src.models.backbones.video_swin.video_swin_aliases import (
+    VideoSwinBBackbone,
+)
+from keras_cv.src.models.backbones.video_swin.video_swin_aliases import (
+    VideoSwinSBackbone,
+)
+from keras_cv.src.models.backbones.video_swin.video_swin_aliases import (
+    VideoSwinTBackbone,
+)
 from keras_cv.src.models.backbones.video_swin.video_swin_backbone import (
     VideoSwinBackbone,
 )

diff --git a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py
@@ -14,6 +14,7 @@
 
 import copy
 
+from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.models.backbones.video_swin.video_swin_backbone import (
     VideoSwinBackbone,
 )
@@ -42,6 +43,7 @@
 """  # noqa: E501
 
 
+@keras_cv_export("keras_cv.models.VideoSwinTBackbone")
 class VideoSwinTBackbone(VideoSwinBackbone):
     def __new__(
         cls,
@@ -79,6 +81,7 @@ def presets_with_weights(cls):
         return cls.presets
 
 
+@keras_cv_export("keras_cv.models.VideoSwinSBackbone")
 class VideoSwinSBackbone(VideoSwinBackbone):
     def __new__(
         cls,
@@ -116,6 +119,7 @@ def presets_with_weights(cls):
         return cls.presets
 
 
+@keras_cv_export("keras_cv.models.VideoSwinBBackbone")
 class VideoSwinBBackbone(VideoSwinBackbone):
     def __new__(
         cls,