generate API

keras-team · Apr 29, 2024 · 41b7b5e · 41b7b5e
1 parent a582cec
commit 41b7b5e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
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,7 +14,7 @@
 
 import copy
 
-from keras_cv.api_export import keras_cv_export
+from keras_cv.src.api_export import keras_cv_export
 from keras_cv.src.models.backbones.video_swin.video_swin_backbone import (
     VideoSwinBackbone,
 )